oj.uz

Submission #59752

# Submission time Handle Problem Language Result Execution time Memory
59752 2018-07-23T04:15:37 Z Benq File Paths (BOI15_fil) C++11
100 / 100
 325 ms 27132 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
 
using namespace std;
using namespace __gnu_pbds;
 
typedef long long ll;
typedef long double ld;
typedef complex<ld> cd;
 
typedef pair<int, int> pi;
typedef pair<ll,ll> pl;
typedef pair<ld,ld> pd;
 
typedef vector<int> vi;
typedef vector<ld> vd;
typedef vector<ll> vl;
typedef vector<pi> vpi;
typedef vector<pl> vpl;
typedef vector<cd> vcd;
 
template <class T> using Tree = tree<T, null_type, less<T>, rb_tree_tag,tree_order_statistics_node_update>;
 
#define FOR(i, a, b) for (int i=a; i<(b); i++)
#define F0R(i, a) for (int i=0; i<(a); i++)
#define FORd(i,a,b) for (int i = (b)-1; i >= a; i--)
#define F0Rd(i,a) for (int i = (a)-1; i >= 0; i--)
 
#define sz(x) (int)(x).size()
#define mp make_pair
#define pb push_back
#define f first
#define s second
#define lb lower_bound
#define ub upper_bound
#define all(x) x.begin(), x.end()
 
const int MOD = 1000000007;
const ll INF = 1e18;
const int MX = 6001;
 
int n,m,k,s,p[MX],l[MX];
ll depth[MX];
vi child[MX];
bitset<MX> ans;
bitset<1000001> root;
 
void dfs0(int x, ll cdepth) {
    cdepth += l[x];
    depth[x] = cdepth;
    for (int i: child[x]) dfs0(i,cdepth);
}
 
void genCyc(int x, int ori, ll cdepth, vi& CYC) {
    if (x > n) return;
    if (cdepth <= k) CYC.pb(cdepth);
    for (int i: child[x]) genCyc(i,ori,cdepth+l[i],CYC);
}

int cyc[1000001];

void DFS(int cur) {
    if (cur > n) {
        ll need = k-depth[cur];
        if (need == 0) {
            ans[cur] = 1;
            return;
        }
        for (int t = 1; t*t <= need; t++) if (need%t == 0) 
            if (cyc[t] || cyc[need/t]) ans[cur] = 1;
    }
    vi CYC; genCyc(cur,cur,s,CYC);
    for (int i: CYC) cyc[i] ++;
    for (int i: child[cur]) DFS(i);
    for (int i: CYC) cyc[i] --;
}
 
void case0() {
    dfs0(0,0);
    DFS(0);
}
 
void case1() {
    F0R(i,n+1) if (depth[i]+s <= k) root[depth[i]+s] = 1;
 
    FOR(i,n+1,n+m+1) {
        vi v = {l[i]};
        int x = p[i];
        while (x) {
            v.pb(v.back()+l[x]);
            x = p[x];
        }
        for (int x: v) if (x <= k && root[k-x]) ans[i] = 1;
    }
}
 
int main() {
    ios_base::sync_with_stdio(0); cin.tie(0);
    cin >> n >> m >> k >> s; s ++;
    FOR(i,1,n+1) {
        cin >> p[i] >> l[i]; l[i] ++;
        child[p[i]].pb(i);
    }
    FOR(i,n+1,n+m+1) {
        cin >> p[i] >> l[i]; l[i] ++;
        child[p[i]].pb(i);
    }
    case0();
    case1();
    FOR(i,n+1,n+m+1) {
        if (ans[i]) cout << "YES";
        else cout << "NO";
        cout << "\n";
    }
}
 
/* Look for:
* the exact constraints (multiple sets are too slow for n=10^6 :( ) 
* special cases (n=1?)
* overflow (ll vs int?)
* array bounds
*/

Subtask #1
33.0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 504 KB Output is correct
2 Correct 3 ms 612 KB Output is correct
3 Correct 4 ms 1160 KB Output is correct
4 Correct 8 ms 1496 KB Output is correct
5 Correct 13 ms 4628 KB Output is correct
6 Correct 12 ms 4628 KB Output is correct
7 Correct 17 ms 4716 KB Output is correct
8 Correct 7 ms 4716 KB Output is correct
9 Correct 8 ms 4716 KB Output is correct
10 Correct 3 ms 4716 KB Output is correct
11 Correct 4 ms 4716 KB Output is correct
12 Correct 9 ms 4716 KB Output is correct

Subtask #2
33.0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 25 ms 4716 KB Output is correct
2 Correct 31 ms 4716 KB Output is correct
3 Correct 31 ms 4856 KB Output is correct
4 Correct 33 ms 4860 KB Output is correct
5 Correct 287 ms 23932 KB Output is correct
6 Correct 244 ms 24460 KB Output is correct
7 Correct 137 ms 24460 KB Output is correct
8 Correct 131 ms 24460 KB Output is correct
9 Correct 28 ms 24460 KB Output is correct
10 Correct 22 ms 24460 KB Output is correct
11 Correct 12 ms 24460 KB Output is correct
12 Correct 208 ms 24460 KB Output is correct

Subtask #3
34.0 / 34.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 504 KB Output is correct
2 Correct 3 ms 612 KB Output is correct
3 Correct 4 ms 1160 KB Output is correct
4 Correct 8 ms 1496 KB Output is correct
5 Correct 13 ms 4628 KB Output is correct
6 Correct 12 ms 4628 KB Output is correct
7 Correct 17 ms 4716 KB Output is correct
8 Correct 7 ms 4716 KB Output is correct
9 Correct 8 ms 4716 KB Output is correct
10 Correct 3 ms 4716 KB Output is correct
11 Correct 4 ms 4716 KB Output is correct
12 Correct 9 ms 4716 KB Output is correct
13 Correct 25 ms 4716 KB Output is correct
14 Correct 31 ms 4716 KB Output is correct
15 Correct 31 ms 4856 KB Output is correct
16 Correct 33 ms 4860 KB Output is correct
17 Correct 287 ms 23932 KB Output is correct
18 Correct 244 ms 24460 KB Output is correct
19 Correct 137 ms 24460 KB Output is correct
20 Correct 131 ms 24460 KB Output is correct
21 Correct 28 ms 24460 KB Output is correct
22 Correct 22 ms 24460 KB Output is correct
23 Correct 12 ms 24460 KB Output is correct
24 Correct 208 ms 24460 KB Output is correct
25 Correct 21 ms 24460 KB Output is correct
26 Correct 20 ms 24460 KB Output is correct
27 Correct 20 ms 24460 KB Output is correct
28 Correct 26 ms 24460 KB Output is correct
29 Correct 325 ms 25016 KB Output is correct
30 Correct 309 ms 25016 KB Output is correct
31 Correct 133 ms 25016 KB Output is correct
32 Correct 191 ms 25016 KB Output is correct
33 Correct 20 ms 25016 KB Output is correct
34 Correct 21 ms 25016 KB Output is correct
35 Correct 14 ms 25016 KB Output is correct
36 Correct 266 ms 27132 KB Output is correct