제출 #714641

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
714641		Khizri	Stranded Far From Home (BOI22_island)	C++17	20 / 100	1081 ms	352852 KiB

island

#pragma GCC optimize ("O3")
#include <bits/stdc++.h>
using namespace std;
//------------------------------DEFINE------------------------------
//******************************************************************
#define IOS ios_base::sync_with_stdio(false); cin.tie(0),cout.tie(0)
#define ll long long
#define int long long
#define pb push_back
#define F first
#define S second
#define INF 1e18
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pii pair<int,int>
#define pll pair<ll,ll>
#define OK cout<<"Ok"<<endl;
#define MOD (ll)(1e9+7)
#define endl "\n"
//******************************************************************
//----------------------------FUNCTION------------------------------
//******************************************************************
ll gcd(ll a,ll b){
    if(a>b) swap(a,b);
    if(a==0) return a+b;
    return gcd(b%a,a);
}
ll lcm(ll a,ll b){
    return a/gcd(a,b)*b;
}
bool is_prime(ll n){
    ll k=sqrt(n);

    if(n==2) return true;
    if(n<2||n%2==0||k*k==n) return false;
    for(int i=3;i<=k;i+=2){
        if(n%i==0){
            return false;
        }
    }
    return true;
}
//*****************************************************************
//--------------------------MAIN-CODE------------------------------
const int mxn=2e5+5;
int t=1,n,m,arr[mxn],sub[mxn],sign[mxn],ans[mxn];
vector<int>vt[mxn];
bool f(int a,int b){
    return arr[a]<arr[b];
}
bool check(int u){
    priority_queue<pii,vector<pii>,greater<pii>>q;
    ll ans=arr[u];
    for(int v:vt[u]){
        q.push({arr[v],v});
    }
    vector<int>color(n+5);
    color[u]=1;
    while(q.size()){
        pii p=q.top();
        q.pop();
        int u=p.S;
        ll sum=p.F;
        if(color[u]) continue;
        if(sum>ans) return false;
        ans+=sum;
        color[u]=1;
        for(int v:vt[u]){
            if(!color[v]){
                q.push({arr[v],v});
            }
        }
    }
    for(int i=1;i<=n;i++){
        if(!color[i]) return false;
    }
    return true;
}
void dfs(int u,int p){
    for(int v:vt[u]){
        if(v!=p){
            dfs(v,u);
            sub[u]+=sub[v];
            if(sub[v]>=arr[u]){
                sign[v]=1;
            }
            else{
                sign[v]=0;
            }
        }
    }
    sub[u]+=arr[u];
}
void dfs2(int u,int p){
    ans[u]=1;
    for(int v:vt[u]){
        if(v!=p){
            if(sign[v]){
                dfs2(v,u);
            }
        }
    }
}
void solve2(){
    dfs(1,-1);
    dfs2(1,-1);
    for(int i=1;i<=n;i++){
        cout<<ans[i];
    }
    cout<<endl;
}
void solve(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>arr[i];
    }
    while(m--){
        int a,b;
        cin>>a>>b;
        vt[a].pb(b);
        vt[b].pb(a);
    }
    if(n>2000){
        solve2();
        return;
    }
    for(int i=1;i<=n;i++){
        if(check(i)){
            cout<<1;
        }
        else{
            cout<<0;
        }
    }
    cout<<endl;
}
signed main(){
    IOS;
    //cin>>t;
    while(t--){
        solve();
    }
    return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 0 / 15
• Subtask 4: 0 / 30
• Subtask 5: 0 / 35