oj.uz

Submission #797798

# Submission time Handle Problem Language Result Execution time Memory
797798 2023-07-29T23:26:34 Z TB_ Stranded Far From Home (BOI22_island) C++17
35 / 100
 1000 ms 32016 KB 
#include <bits/stdc++.h>
using namespace std;

// #undef _GLIBCXX_DEBUG                // disable run-time bound checking, etc
// #pragma GCC optimize("Ofast,inline") // Ofast = O3,fast-math,allow-store-data-races,no-protect-parens
// #pragma GCC optimize ("unroll-loops")

// #pragma GCC target("bmi,bmi2,lzcnt,popcnt")                      // bit manipulation
// #pragma GCC target("movbe")                                      // byte swap
// #pragma GCC target("aes,pclmul,rdrnd")                           // encryption
// #pragma GCC target("avx,avx2,f16c,fma,sse3,ssse3,sse4.1,sse4.2") // SIMD

// #include <bits/extc++.h>
// using namespace __gnu_pbds;
// template<class T>using ordered_set = tree<T, null_type, less<T>, rb_tree_tag,tree_order_statistics_node_update>;
// template<class T>using ordered_multiset = tree<T, null_type, less_equal<T>, rb_tree_tag,tree_order_statistics_node_update>;

#define ll long long
#define INF (ll)1e9+7
#define fo(i, n) for(int i=0;i<((ll)n);i++)
#define deb(x) cout << #x << " = " << x << endl;
#define deb2(x, y) cout << #x << " = " << x << ", " << #y << " = " << y << endl
#define pb push_back
#define mp make_pair
#define F first
#define S second
#define LSOne(S) ((S) & (-S))
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
inline int readint(){ int v = 0; char c; while((c = getchar()) != EOF && c != ' ' && c != '\n'){ v *= 10; v += c - '0'; } return v; }
inline int readintsigned() { int v = 0; int sign = 1; char c = getchar(); if (c == '-') { sign = -1; } else { v += c - '0'; } while ((c = getchar()) != EOF && c != ' ' && c != '\n') { v *= 10; v += c - '0'; } return v * sign; }
inline string readstring() { string s; char c; while ((c = getchar()) != EOF && c != '\n' && c != ' ') { s.push_back(c); } return s; }
template <class result_t=std::chrono::milliseconds,class clock_t=std::chrono::steady_clock,class duration_t = std::chrono::milliseconds>
auto since(std::chrono::time_point<clock_t, duration_t> const& start){return std::chrono::duration_cast<result_t>(clock_t::now() - start);}
typedef pair<int, int> pii;
typedef pair<ll, ll> pl;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef vector<pii> vpii;
typedef vector<pl> vpl;
typedef vector<vi> vvi;
typedef vector<vl> vvl;
typedef vector<vpii> vvpii;
typedef vector<vpl> vvpl;

ll n, m, val, from, to, w;
vl population;

vvl adj(200001);

struct UnionFind{
    int n;
    vl p, siz;
    UnionFind(int n) : n(n){
        p.assign(n, -1);
        siz.assign(n, 1);
        fo(i, n)p[i] = i;
    }

    int find(int x){
        if(p[x] == x) return x;
        return p[x] = find(p[x]);
    }

    void unite(int a, int b){
        a = find(a);
        b = find(b);
        if(a == b) return;
        if(siz[b] > siz[a]) swap(a, b);
        siz[a]+=siz[b];
        p[b] = a;
    }
};

int main() {
    // cout << fixed << setprecision(20);
    // auto start = std::chrono::steady_clock::now(); // since(start).count()
    cin.tie(0)->sync_with_stdio(0);
    mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

    cin >> n >> m;
    pii biggestPop = {-1, -1};
    fo(i, n){
        cin >> val;
        population.pb(val);
        biggestPop = max(biggestPop, {val, i});
    }
    fo(i, m){
        cin >> from >> to;
        adj[--from].pb(--to);
        adj[to].pb(from);
    }
    vi ans(n, -1);
    ans[biggestPop.S] = 1;

    vi order;
    fo(i, n) order.pb(i);
    shuffle(all(order), rng);

    UnionFind uf(n);

    vi seen(n, 0);
    for(auto start : order){
        start = uf.find(start);
        if(ans[start] != -1) continue;
        vi visited;
        set<pl> s = {{0, start}};

        ll currentPop = population[start];
        while(!s.empty()){
            auto res = s.lower_bound({currentPop, INF});
            if(res == s.begin()) {
                if(seen[uf.find((*res).S)]) continue;
                for(int i = 1; i < visited.size(); i++){
                    uf.unite(visited[0], visited[i]);
                    for(auto v : adj[visited[i]]){
                        if(seen[v]) continue;
                        seen[v] = 1;
                        adj[visited[0]].pb(v);
                    }
                }
                population[visited[0]] = currentPop;
                for(auto v : adj[visited[0]]) seen[v] = 0;
                visited = {visited[0]};
                seen[visited[0]] = 1;
                ans[visited[0]] = 0;
                start = uf.find((*res).S);
                res++;
            }


            tie(w, to) = *--res;
            to = uf.find(to);
            s.erase(res);
            if(seen[to]) continue;
            seen[to] = 1;
            visited.pb(to);
            currentPop += w;
            if(ans[to] == 1 || currentPop >= biggestPop.F) break;
            for(auto v : adj[to]){
                v = uf.find(v);
                s.insert({population[v], v});           
            }
        }
        ans[start] = 1;
        for(auto v : visited) 
            seen[uf.find(v)] = 0;
        
    }
    fo(i, n) cout << ans[uf.find(i)];

    return 0;
}

Compilation message

island.cpp: In function 'int main()':
island.cpp:114:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  114 |                 for(int i = 1; i < visited.size(); i++){
      |                                ~~^~~~~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4948 KB Output is correct
2 Correct 3 ms 4948 KB Output is correct
3 Correct 3 ms 4948 KB Output is correct
4 Correct 4 ms 5056 KB Output is correct
5 Correct 4 ms 5204 KB Output is correct
6 Correct 3 ms 5164 KB Output is correct
7 Correct 3 ms 5172 KB Output is correct
8 Correct 4 ms 5172 KB Output is correct
9 Correct 3 ms 5204 KB Output is correct
10 Correct 3 ms 5168 KB Output is correct
11 Correct 4 ms 5204 KB Output is correct
12 Correct 3 ms 5152 KB Output is correct
13 Correct 3 ms 5204 KB Output is correct
14 Correct 4 ms 5204 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 2 ms 4948 KB Output is correct
3 Correct 173 ms 19708 KB Output is correct
4 Correct 145 ms 24132 KB Output is correct
5 Correct 150 ms 19428 KB Output is correct
6 Correct 160 ms 19852 KB Output is correct
7 Correct 163 ms 19960 KB Output is correct
8 Correct 159 ms 19872 KB Output is correct
9 Correct 162 ms 21196 KB Output is correct
10 Correct 125 ms 20492 KB Output is correct
11 Correct 85 ms 20436 KB Output is correct
12 Correct 146 ms 19908 KB Output is correct
13 Correct 156 ms 30652 KB Output is correct
14 Correct 117 ms 23940 KB Output is correct
15 Correct 179 ms 18824 KB Output is correct
16 Correct 357 ms 20260 KB Output is correct

Subtask #3
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 165 ms 19464 KB Output is correct
3 Correct 161 ms 23828 KB Output is correct
4 Correct 102 ms 23164 KB Output is correct
5 Correct 117 ms 24196 KB Output is correct
6 Correct 151 ms 23828 KB Output is correct
7 Correct 148 ms 23172 KB Output is correct
8 Correct 143 ms 23232 KB Output is correct
9 Correct 373 ms 23880 KB Output is correct
10 Correct 413 ms 22888 KB Output is correct
11 Correct 147 ms 24432 KB Output is correct

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 4948 KB Output is correct
2 Correct 174 ms 20100 KB Output is correct
3 Correct 165 ms 24216 KB Output is correct
4 Correct 192 ms 24188 KB Output is correct
5 Correct 104 ms 24148 KB Output is correct
6 Correct 101 ms 23584 KB Output is correct
7 Correct 123 ms 24716 KB Output is correct
8 Execution timed out 1092 ms 32016 KB Time limit exceeded
9 Halted 0 ms 0 KB -

Subtask #5
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 4948 KB Output is correct
2 Correct 3 ms 4948 KB Output is correct
3 Correct 3 ms 4948 KB Output is correct
4 Correct 4 ms 5056 KB Output is correct
5 Correct 4 ms 5204 KB Output is correct
6 Correct 3 ms 5164 KB Output is correct
7 Correct 3 ms 5172 KB Output is correct
8 Correct 4 ms 5172 KB Output is correct
9 Correct 3 ms 5204 KB Output is correct
10 Correct 3 ms 5168 KB Output is correct
11 Correct 4 ms 5204 KB Output is correct
12 Correct 3 ms 5152 KB Output is correct
13 Correct 3 ms 5204 KB Output is correct
14 Correct 4 ms 5204 KB Output is correct
15 Correct 2 ms 4948 KB Output is correct
16 Correct 2 ms 4948 KB Output is correct
17 Correct 173 ms 19708 KB Output is correct
18 Correct 145 ms 24132 KB Output is correct
19 Correct 150 ms 19428 KB Output is correct
20 Correct 160 ms 19852 KB Output is correct
21 Correct 163 ms 19960 KB Output is correct
22 Correct 159 ms 19872 KB Output is correct
23 Correct 162 ms 21196 KB Output is correct
24 Correct 125 ms 20492 KB Output is correct
25 Correct 85 ms 20436 KB Output is correct
26 Correct 146 ms 19908 KB Output is correct
27 Correct 156 ms 30652 KB Output is correct
28 Correct 117 ms 23940 KB Output is correct
29 Correct 179 ms 18824 KB Output is correct
30 Correct 357 ms 20260 KB Output is correct
31 Correct 2 ms 4948 KB Output is correct
32 Correct 165 ms 19464 KB Output is correct
33 Correct 161 ms 23828 KB Output is correct
34 Correct 102 ms 23164 KB Output is correct
35 Correct 117 ms 24196 KB Output is correct
36 Correct 151 ms 23828 KB Output is correct
37 Correct 148 ms 23172 KB Output is correct
38 Correct 143 ms 23232 KB Output is correct
39 Correct 373 ms 23880 KB Output is correct
40 Correct 413 ms 22888 KB Output is correct
41 Correct 147 ms 24432 KB Output is correct
42 Correct 2 ms 4948 KB Output is correct
43 Correct 174 ms 20100 KB Output is correct
44 Correct 165 ms 24216 KB Output is correct
45 Correct 192 ms 24188 KB Output is correct
46 Correct 104 ms 24148 KB Output is correct
47 Correct 101 ms 23584 KB Output is correct
48 Correct 123 ms 24716 KB Output is correct
49 Execution timed out 1092 ms 32016 KB Time limit exceeded
50 Halted 0 ms 0 KB -