Submission #1092677

#	Time	Username	Problem	Language	Result	Execution time	Memory
1092677		vjudge1	Regions (IOI09_regions)	C++17	45 / 100	8093 ms	131072 KiB

regions

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#pragma GCC target("popcnt")
using namespace std;
 
using ll = long long;
using ull = unsigned long long;
using lld = long double;
using ii = pair<int,int>;
using pll = pair<ll, ll>;
 
using vi = vector<int>;
using vll = vector<ll>;
using vii = vector<ii>;
using vpll = vector<pll>;
using vlld = vector<lld>;
 
// #define endl '\n'
#define all(x) x.begin(),x.end()
#define lsb(x) x&(-x)
#define gcd(a,b) __gcd(a,b)
#define sz(x) (int)x.size()
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define fls cout.flush()
 
#define fore(i,l,r) for(auto i=l;i<r;i++)
#define fo(i,n) fore(i,0,n)
#define forex(i,r,l) for(auto i=r; i>=l;i--)
#define ffo(i,n) forex(i,n-1,0)
 
bool cmin(ll &a, ll b) { if(b<a){a=b;return 1;}return 0; }
bool cmax(ll &a, ll b) { if(b>a){a=b;return 1;}return 0; }
void valid(ll in) { cout<<((in)?"YES\n":"NO\n"); }
ll lcm(ll a, ll b) { return (a/gcd(a,b))*b; }
ll gauss(ll n) { return (n*(n+1))/2; }

struct SegTree{
    SegTree *left, *right;
    int l,r,sum=0;
    SegTree(){}
    SegTree(int l, int r):l(l),r(r){}
    void build(){
        if(l==r)sum=0;
        else{
            int mid = (l+r)/2;
            left = new SegTree(l, mid);
            right = new SegTree(mid+1, r);
            left->build();right->build();
        }
    }
    SegTree *update(int i, int v){
        if(r<i || l>i)return this;
        SegTree *t = new SegTree(l, r);
        if(l==r)t->sum=sum+v;
        else{
            t->left=left->update(i, v);
            t->right=right->update(i,v);
            t->sum=t->left->sum+t->right->sum;
        }
        return t;
    }
    int query(int i, int j){
        if(l>j||r<i)return 0;
        if(l>=i&&r<=j)return sum;
        return right->query(i, j)+left->query(i,j);
    }
};
const int N = 2e5 + 7;
int pa[N], h[N], el[N], tin[N], tout[N], timer, n, r, Q;
vi graph[N];
map<int, int> rd[N];
SegTree *st[N];
void test_case() {
    cin >> n >> r >> Q;
    vector<vi> ord(r+1); 
    cin >> h[1];
    fore (i, 2, n+1) {
        cin >> pa[i];
        graph[pa[i]].pb(i);
        cin >> h[i];
    }
    fore (i, 1, n+1) ord[h[i]].pb(i);
    if (r <= 500) {
        vector<vi> dp(501, vi(501, 0));
        function<void(int)> dfs = [&](int u) {
            rd[u][h[u]]++;
            for(int v: graph[u]) {
                dfs(v);
                if(sz(rd[v]) > sz(rd[u])) swap(rd[u], rd[v]);
                for (auto [k, vl]: rd[v]) {
                    rd[u][k] += vl;
                }
            }
            for (auto [k, vl]: rd[u]) dp[h[u]][k] += vl;
        };
        dfs(1);
        while (Q--) {
            int r1, r2;
            cin >> r1 >> r2;
            cout << dp[r1][r2] << endl;
        }
        return;
    }
     int mx = 0;
    // fore (i, 1, r+1) mx = max(mx, sz(ord[i])); 
    // assert(mx <= 500);
    timer = 1;
    function<void(int)> dfs = [&](int u) {
        el[timer]=u;
        tin[u] = timer++;
        for(int v: graph[u]) dfs(v);
        tout[u] = timer-1;
    };
    dfs(1);
    st[0] = new SegTree(1, r);
    st[0]->build();
    fore (i, 1, n+1) st[i] = st[i-1]->update(h[el[i]], +1);
    while (Q--) {
        int r1, r2;
        cin >> r1 >> r2;
        int ans = 0;
        for(int e1: ord[r1]) {
            ans += st[tout[e1]]->query(r2, r2) - st[tin[e1]-1]->query(r2, r2);
        }
        cout << ans << endl;
    }
} 

int main() {
    int tt = 1;
    // cin >> tt;
    while(tt--) test_case();
}

Compilation message (stderr)

regions.cpp:4: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    4 | #pragma GCC optimization ("O3")
      | 
regions.cpp:5: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    5 | #pragma GCC optimization ("unroll-loops")
      | 
regions.cpp: In function 'void test_case()':
regions.cpp:110:10: warning: unused variable 'mx' [-Wunused-variable]
  110 |      int mx = 0;
      |          ^~

• Subtask 1: 15 / 15
• Subtask 2: 30 / 85