Submission #897043

# Submission time Handle Problem Language Result Execution time Memory
897043 2024-01-02T13:28:51 Z bLIC Regions (IOI09_regions) C++17
13 / 100
3324 ms 37836 KB
/**
 *  Author: Anil Byar
**/

#include <bits/stdc++.h>
// #include <ext/pb_ds/assoc_container.hpp>
// #include <ext/pb_ds/tree_policy.hpp>

// using namespace __gnu_pbds;
using namespace std;

// template<class T>
// using ordered_set = tree<T, null_type, less_equal<T>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;



#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define sz(x) (int)(x).size()
#define uniq(v) v.erase(unique(all(v)), v.end())
#define ft first
#define sd second
#define pb push_back
#define eb emplace_back

// Source: https://codeforces.com/blog/entry/68809
// { start
void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(long long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '"' << x << '"';}
void __print(const string &x) {cerr << '"' << x << '"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? "," : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#ifndef ONLINE_JUDGE
#define debug(x...) cerr << "[" << #x << "] = ["; _print(x)
#else
#define debug(x...)
#endif
// } end


typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;
typedef vector<pii> vii;
typedef vector<ll> vl;
typedef vector<pll> vll;
typedef vector<vi> vvi;
typedef vector<vii> vvii;
typedef vector<vl> vvl;

#define dbg if(0)

const ll MOD = 1e9+7;
const ll MOD9 = 998244353;
const ll INFLL = 1e18+5;
const int INF = 1e9;

void printbit(int x, int len) {string s="\n";while(len--){s=((x%2)?'1':'0')+s;x/=2;} cout<<s;}
ll power(ll x, ll y, ll mod){if (y==0) return 1;ll ret = power(x, y/2, mod);ret *= ret;ret%=mod;if (y&1) ret *= x;return ret%mod;}
ll modinv(ll x, ll mod = MOD) {return power(x, mod-2, mod);}

template<class T> bool chmin(T& a, T b){return (a>b?a=b,1:0);}
template<class T> bool chmax(T& a, T b){return (a<b?a=b,1:0);}
template<class T>
istream& operator>>(istream&in, vector<T>&v){
    for (T& x:v) in>>x;
    return in;
}
template<class T>
ostream& operator<<(ostream&out, vector<T>&v){
    for (T& x:v) out<<x<<' ';
    cout<<'\n';
    return out;
}
// use ?: with brackets (?:)
// check for overflow
// think about dp
// Read the statement carefully

const int N = 2e5+1, R = 25000;
const int B = 450;

vi adj[N], emp[R], tour[R];
int reg[N];
int cntreg[R], idofreg[R];
int cntans[B+10][R];
int currcnt[R];
vector<int> anc;
int st[N], en[N], timer;

void dfs(int node){
    st[node] = timer++;
    tour[reg[node]].push_back(st[node]);
    int id = idofreg[reg[node]];
    if (id && currcnt[id]++==0) anc.push_back(id);
    for (int x:adj[node]){
        for (int y:anc) cntans[y][reg[x]] += currcnt[y];
        dfs(x); 
    }
    if (id && --currcnt[id]==0) anc.pop_back();
    en[node] = timer++;
}


void solve(){
    int n, r, q;cin>>n>>r>>q;
    {
        int x;cin>>x;
        cntreg[x]++;
        x--;
        emp[x].push_back(0);
        reg[0] = x;
    }
    for (int i = 1;i<n;i++){
        int p, x;cin>>p>>x; x--;p--;
        cntreg[x]++;
        reg[i] = x;
        emp[x].push_back(i);
        adj[p].push_back(i);
    }
    int rg = 0;
    for (int i = 0;i<r;i++){
        if (cntreg[i]>=B) idofreg[i] = ++rg;
    }
    dfs(0);
    while(q--){
        int r1, r2;cin>>r1>>r2;
        r1--;
        r2--;
        if (idofreg[r1]) cout<<cntans[idofreg[r1]][r2];
        else {
            ll ans = 0;
            for (int e:emp[r1]){
                ans += lower_bound(all(tour[r2]), en[e]) - upper_bound(all(tour[r2]), st[e]);
            }
            cout<<ans;
        }
        cout<<endl;
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);


#define ONLINE_JUDGE
#ifndef ONLINE_JUDGE
    freopen("input.in", "r", stdin);
    freopen("output.out", "w", stdout);
    freopen("debug.dbg", "w", stderr);
    int tt = clock();
#endif

    int t=1, i = 1;
    // cin>>t;
    while(t--){
        // cout<<"Case #"<<i++<<": ";
        solve();
        cout<<'\n';
    }
#ifndef ONLINE_JUDGE
    cerr << "TIME = " << (float)(clock() - tt)/CLOCKS_PER_SEC << endl;
    tt = clock();
#endif
}

Compilation message

regions.cpp: In function 'int main()':
regions.cpp:173:14: warning: unused variable 'i' [-Wunused-variable]
  173 |     int t=1, i = 1;
      |              ^
# Verdict Execution time Memory Grader output
1 Correct 2 ms 10584 KB Output is correct
2 Correct 2 ms 10584 KB Output is correct
3 Correct 2 ms 10584 KB Output is correct
4 Correct 3 ms 10584 KB Output is correct
5 Runtime error 5 ms 10748 KB Execution killed with signal 13
6 Runtime error 10 ms 10836 KB Execution killed with signal 13
7 Correct 15 ms 10584 KB Output is correct
8 Correct 17 ms 10840 KB Output is correct
9 Correct 25 ms 11352 KB Output is correct
10 Runtime error 50 ms 11196 KB Execution killed with signal 13
11 Runtime error 69 ms 11444 KB Execution killed with signal 13
12 Correct 80 ms 12032 KB Output is correct
13 Runtime error 120 ms 11536 KB Execution killed with signal 13
14 Runtime error 197 ms 12092 KB Execution killed with signal 13
15 Runtime error 185 ms 16412 KB Execution killed with signal 13
# Verdict Execution time Memory Grader output
1 Runtime error 1346 ms 17380 KB Execution killed with signal 13
2 Runtime error 1586 ms 15636 KB Execution killed with signal 13
3 Runtime error 2043 ms 20120 KB Execution killed with signal 13
4 Runtime error 176 ms 12240 KB Execution killed with signal 13
5 Runtime error 205 ms 14888 KB Execution killed with signal 13
6 Runtime error 305 ms 19596 KB Execution killed with signal 13
7 Runtime error 1137 ms 16292 KB Execution killed with signal 13
8 Correct 621 ms 32272 KB Output is correct
9 Runtime error 1770 ms 19444 KB Execution killed with signal 13
10 Runtime error 2667 ms 33592 KB Execution killed with signal 13
11 Runtime error 3324 ms 17968 KB Execution killed with signal 13
12 Runtime error 892 ms 19472 KB Execution killed with signal 13
13 Runtime error 1287 ms 20400 KB Execution killed with signal 13
14 Runtime error 1443 ms 21676 KB Execution killed with signal 13
15 Runtime error 2213 ms 26896 KB Execution killed with signal 13
16 Runtime error 2097 ms 37836 KB Execution killed with signal 13
17 Runtime error 1993 ms 37536 KB Execution killed with signal 13