oj.uz

Submission #723286

# Submission time Handle Problem Language Result Execution time Memory
723286 2023-04-13T13:54:09 Z MurotY Stranded Far From Home (BOI22_island) C++14
10 / 100
 1000 ms 233280 KB 
#pragma GCC optimize("Ofast")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target("avx,avx2")
#include <bits/stdc++.h>
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define ll long long
#define ull unsigned long long
#define ff first
#define ss second
#define all(a) a.begin(), a.end()
#define sz size()
using namespace std;
const double pi = 2 * acos(0.0);
const ll N=1e6+7, M=998244353;
vector <ll> g[N];
ll a[N], res[N];  
bool u[N];
ll sum[N], ans[N];
ll dfs(int v, int p){
    sum[v]=a[v];
    for (auto l:g[v]){
        if (l != p){
            sum[v]+=dfs(l, v);
        }
    }
    return sum[v];
}

void dfs1(int v, int p){
    if (sum[v] >= a[p]) ans[v]=1;
    if (!ans[v]) return ;
    u[v]=1;
    for (auto l:g[v]){
        if (l != p){
            dfs1(l, v);
        }
    }
    return ;
}
void solve()
{
	int n, m;
	cin >> n >> m;
	
	for (int i=1;i<=n;i++) cin >> a[i];
	for (int i=1;i<=m;i++){
		int x, y;
		cin >> x >> y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
    if (n <= 2000 && m <= 2000){
		for (int i=1;i<=n;i++){
			set <pair <ll, ll>> q;
			vector <int> u(n+5, 0);
			q.insert({0, i});
			ll sum=0, cnt=0;
			while (!q.empty()){
			    pair <ll,ll> mn=*q.begin();
			    if (sum < mn.ff) break;
			    if (u[mn.ss]) continue;
			    u[mn.ss]=1;
			    q.erase(mn);
			    sum+=a[mn.ss];
			    for (auto l:g[mn.ss]){
			        if (!u[l]) {
			            q.insert({a[l], l});
			        }
			    }
			}
			if (q.sz == 0) cout << "1";
			else cout << "0";
		}
		return ;
    }
    ans[1]=1;
    dfs(1, 1);
    dfs1(1, 1);
    for (int i=1;i<=n;i++) cout << ans[i] << " ";
    return;
}
int main(){
	ios;
	int t=1;	
//	cin >> t;
	while (t--){ 
	    solve();
	    cout << "\n";
	}
	return 0;
}

Compilation message

island.cpp: In function 'void solve()':
island.cpp:57:14: warning: unused variable 'cnt' [-Wunused-variable]
   57 |    ll sum=0, cnt=0;
      |              ^~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 23764 KB Output is correct
2 Correct 16 ms 23756 KB Output is correct
3 Correct 13 ms 23808 KB Output is correct
4 Correct 320 ms 23952 KB Output is correct
5 Correct 268 ms 23944 KB Output is correct
6 Correct 456 ms 24012 KB Output is correct
7 Correct 295 ms 23892 KB Output is correct
8 Correct 231 ms 23936 KB Output is correct
9 Correct 492 ms 23992 KB Output is correct
10 Correct 131 ms 23852 KB Output is correct
11 Correct 123 ms 23928 KB Output is correct
12 Correct 156 ms 23948 KB Output is correct
13 Correct 249 ms 23912 KB Output is correct
14 Correct 155 ms 23920 KB Output is correct

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 23764 KB Output is correct
2 Correct 14 ms 23764 KB Output is correct
3 Incorrect 169 ms 39464 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 23764 KB Output is correct
2 Incorrect 153 ms 42700 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 23756 KB Output is correct
2 Execution timed out 1088 ms 233280 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #5
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 14 ms 23764 KB Output is correct
2 Correct 16 ms 23756 KB Output is correct
3 Correct 13 ms 23808 KB Output is correct
4 Correct 320 ms 23952 KB Output is correct
5 Correct 268 ms 23944 KB Output is correct
6 Correct 456 ms 24012 KB Output is correct
7 Correct 295 ms 23892 KB Output is correct
8 Correct 231 ms 23936 KB Output is correct
9 Correct 492 ms 23992 KB Output is correct
10 Correct 131 ms 23852 KB Output is correct
11 Correct 123 ms 23928 KB Output is correct
12 Correct 156 ms 23948 KB Output is correct
13 Correct 249 ms 23912 KB Output is correct
14 Correct 155 ms 23920 KB Output is correct
15 Correct 13 ms 23764 KB Output is correct
16 Correct 14 ms 23764 KB Output is correct
17 Incorrect 169 ms 39464 KB Output isn't correct
18 Halted 0 ms 0 KB -