Submission #970278

#TimeUsernameProblemLanguageResultExecution timeMemory
970278VMaksimoski008Pilot (NOI19_pilot)C++17
89 / 100
1087 ms122892 KiB
#include <bits/stdc++.h>
using namespace std;
 
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx,bmi,bmi2,lzcnt,popcnt")
 
using ll = long long;
using pii = pair<int, int>;
using Query = array<int, 4>;
 
const int maxn = 1e6 + 5;
 
int par[maxn], sz[maxn];

void init(int n) {
    for(int i=0; i<n; i++) par[i] = i, sz[i] = 1;
}

int get(int x) {
    if(x == par[x]) return x;
    return par[x] = get(par[x]);
}

bool uni(int a, int b) {
    a = get(a);
    b = get(b);

    if(a == b) return 1;
    if(sz[a] < sz[b]) swap(a, b);
    sz[a] += sz[b];
    par[b] = a;

    return 1;
}

bool sameSet(int u, int v) { return get(u) == get(v); }
int getSize(int u) { return sz[get(u)]; }
 
ll C(ll x) {
    return x * (x - 1) / 2;
}

vector<int> mp[maxn+1];
 
int32_t main() {
    ios_base::sync_with_stdio(false);
    cout.tie(0); cin.tie(0);
    
    int n, q, mn = 1e9;
    cin >> n >> q;
 
    vector<int> v(n);
    vector<Query> qus;
 
    for(int &x : v) cin >> x;
 
    for(int i=1; i<n; i++)
        qus.push_back({ 1, i-1, i, max(v[i], v[i-1]) });
    
    for(int i=0; i<q; i++) {
        int x;
        cin >> x;
        mn = min(mn, x);
        qus.push_back({ 2, i, i, x });
    }
 
    sort(qus.begin(), qus.end(), [&](Query &a, Query &b) {
        if(a[3] == b[3]) return a[0] < b[0];
        return a[3] < b[3];
    });

    vector<ll> ans(q);
    ll res = 0;
 
    for(int i=0; i<n; i++) mp[v[i]].push_back(i);
 
    set<int> s;
    for(int &x : v) s.insert(x);
    vector<int> comp(s.begin(), s.end());
 
    int ptr = 0;
    init(n);
 
    for(Query &qu : qus) {
        while(ptr < comp.size() && comp[ptr] <= qu[3]) {
            for(int &p : mp[comp[ptr]]) res++;
            ptr++;
        }
 
        if(qu[0] == 1) {
            //if(sameSet(qu[1], qu[2])) continue;
            res -= C(getSize(qu[1]));
            res -= C(getSize(qu[2]));
            uni(qu[1], qu[2]);
            res += C(getSize(qu[1]));
        } else {
            ans[qu[1]] = res;
        }
    }
 
    for(ll &x : ans) cout << x << '\n';
    return 0;
}

Compilation message (stderr)

pilot.cpp: In function 'int32_t main()':
pilot.cpp:85:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   85 |         while(ptr < comp.size() && comp[ptr] <= qu[3]) {
      |               ~~~~^~~~~~~~~~~~~
pilot.cpp:86:22: warning: unused variable 'p' [-Wunused-variable]
   86 |             for(int &p : mp[comp[ptr]]) res++;
      |                      ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...