oj.uz

Submission #993651

# Submission time Handle Problem Language Result Execution time Memory
993651 2024-06-06T09:30:54 Z amine_aroua Stranded Far From Home (BOI22_island) C++17
10 / 100
 1000 ms 23088 KB 
#include<bits/stdc++.h>
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
using namespace std;
#define int long long
#define pb push_back
#define fore(i , n) for(int i = 0 ; i<n;i++)
#define forr(i , x , y) for(int i = x ; i <= y; i++)
#define forn(i , x , y) for(int i = x ; i >= y; i--)
const int N = 2e5 + 10;
vector<int> adj[N];
int a[N];
int n , m;
vector<int> ans(N , -1);
void expand(int x)
{
    priority_queue<pair<int ,int> , vector<pair<int ,int>>  , greater<pair<int , int>>> pq;
    int cur = a[x];
    vector<bool> vis(n , 0);
    pq.push({cur , x});
    vector<int> possible;
    for(auto u : adj[x])
    {
        if(a[u] == a[x])
            possible.pb(u);
    }
    while(!pq.empty())
    {
        auto [val , node] = pq.top();
        pq.pop();
        if(val == cur)
        {
            possible.pb(node);
        }
        if(val > cur)
            break;
        if(vis[node])
            continue;
        vis[node] = 1;
        if(node != x)
            cur+=val;
        if(ans[node] == 1)
        {
            ans[x] = 1;
            return;
        }
        for(auto u : adj[node])
        {
            pq.push({a[u] , u});
        }
    }
    bool acc = 1;
    fore(i , n)
    {
        if(!vis[i])
        {
            acc = 0;
            break;
        }
    }
    if(acc)
    {
        for(auto u : possible)
            ans[u] = 1;
        ans[x] = 1;
    }
    else
    {
        fore(i , n)
        {
            if(vis[i])
                ans[i] = 0;
        }
        ans[x] = 0;
    }
}
signed main()
{

    n = 2e5 , m = n - 1;
    cin>>n>>m;
    vector<pair<int ,int>> ve;
    fore(i , n)
    {

        a[i] = 1;
        cin>>a[i];
        ve.pb({a[i] , i});
    }
    sort(ve.begin() , ve.end());
    fore(i , m)
    {
        int u , v;
        u = i + 1 ;
        v = i + 2;
        cin>>u>>v;
        u-- , v--;
        adj[u].pb(v);
        adj[v].pb(u);
    }
    for(auto [val , i] : ve)
    {
        if(ans[i] == -1)
            expand(i);
    }
    fore(i , n)
    {
        cout<<ans[i];
    }
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 6488 KB Output is correct
2 Correct 3 ms 6492 KB Output is correct
3 Correct 3 ms 6492 KB Output is correct
4 Correct 5 ms 6748 KB Output is correct
5 Correct 6 ms 6732 KB Output is correct
6 Correct 5 ms 6748 KB Output is correct
7 Correct 6 ms 6868 KB Output is correct
8 Correct 5 ms 6724 KB Output is correct
9 Correct 4 ms 6748 KB Output is correct
10 Correct 6 ms 7000 KB Output is correct
11 Correct 7 ms 6748 KB Output is correct
12 Correct 6 ms 6732 KB Output is correct
13 Correct 4 ms 6748 KB Output is correct
14 Correct 13 ms 6832 KB Output is correct

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 6492 KB Output is correct
2 Correct 3 ms 6568 KB Output is correct
3 Execution timed out 1082 ms 23028 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #3
0 / 15.0

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

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 6492 KB Output is correct
2 Execution timed out 1086 ms 23088 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #5
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 6488 KB Output is correct
2 Correct 3 ms 6492 KB Output is correct
3 Correct 3 ms 6492 KB Output is correct
4 Correct 5 ms 6748 KB Output is correct
5 Correct 6 ms 6732 KB Output is correct
6 Correct 5 ms 6748 KB Output is correct
7 Correct 6 ms 6868 KB Output is correct
8 Correct 5 ms 6724 KB Output is correct
9 Correct 4 ms 6748 KB Output is correct
10 Correct 6 ms 7000 KB Output is correct
11 Correct 7 ms 6748 KB Output is correct
12 Correct 6 ms 6732 KB Output is correct
13 Correct 4 ms 6748 KB Output is correct
14 Correct 13 ms 6832 KB Output is correct
15 Correct 3 ms 6492 KB Output is correct
16 Correct 3 ms 6568 KB Output is correct
17 Execution timed out 1082 ms 23028 KB Time limit exceeded
18 Halted 0 ms 0 KB -