Submission #897082

# Submission time Handle Problem Language Result Execution time Memory
897082 2024-01-02T14:16:20 Z bLIC Regions (IOI09_regions) C++17
12 / 100
3314 ms 42520 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];
long long 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;
        x--;
        cntreg[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;
        debug(i, idofreg[i]);
    }
    // assert(rg<=B);
    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:175:14: warning: unused variable 'i' [-Wunused-variable]
  175 |     int t=1, i = 1;
      |              ^
# Verdict Execution time Memory Grader output
1 Correct 2 ms 10328 KB Output is correct
2 Correct 3 ms 10328 KB Output is correct
3 Correct 3 ms 10328 KB Output is correct
4 Correct 4 ms 10328 KB Output is correct
5 Correct 6 ms 10584 KB Output is correct
6 Correct 11 ms 10616 KB Output is correct
7 Correct 14 ms 10584 KB Output is correct
8 Runtime error 20 ms 10704 KB Execution killed with signal 13
9 Runtime error 30 ms 11300 KB Execution killed with signal 13
10 Runtime error 59 ms 11068 KB Execution killed with signal 13
11 Runtime error 75 ms 11332 KB Execution killed with signal 13
12 Runtime error 100 ms 11956 KB Execution killed with signal 13
13 Runtime error 124 ms 11416 KB Execution killed with signal 13
14 Runtime error 226 ms 11964 KB Execution killed with signal 13
15 Runtime error 192 ms 16292 KB Execution killed with signal 13
# Verdict Execution time Memory Grader output
1 Runtime error 1317 ms 19300 KB Execution killed with signal 13
2 Runtime error 1558 ms 15428 KB Execution killed with signal 13
3 Runtime error 2092 ms 19824 KB Execution killed with signal 13
4 Correct 191 ms 11968 KB Output is correct
5 Runtime error 249 ms 14884 KB Execution killed with signal 13
6 Runtime error 341 ms 25808 KB Execution killed with signal 13
7 Runtime error 1119 ms 18740 KB Execution killed with signal 13
8 Runtime error 710 ms 42520 KB Execution killed with signal 13
9 Runtime error 1685 ms 19800 KB Execution killed with signal 13
10 Runtime error 2655 ms 38288 KB Execution killed with signal 13
11 Runtime error 3314 ms 18556 KB Execution killed with signal 13
12 Runtime error 1149 ms 19792 KB Execution killed with signal 13
13 Runtime error 1456 ms 20720 KB Execution killed with signal 13
14 Runtime error 1668 ms 24156 KB Execution killed with signal 13
15 Runtime error 2386 ms 29528 KB Execution killed with signal 13
16 Runtime error 2415 ms 40476 KB Execution killed with signal 13
17 Runtime error 2249 ms 40024 KB Execution killed with signal 13