QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#100964#5830. 树Scintilla20 3703ms612884kbC++143.4kb2023-04-28 21:06:362023-04-28 21:06:37

Judging History

你现在查看的是最新测评结果

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-04-28 21:06:37]
  • 评测
  • 测评结果:20
  • 用时:3703ms
  • 内存:612884kb
  • [2023-04-28 21:06:36]
  • 提交

answer

#include <bits/stdc++.h>

using namespace std;

#define int long long
#define rep(i, s, e) for (int i = s; i <= e; ++i)
#define drep(i, s, e) for (int i = s; i >= e; --i)
#define file(a) freopen(#a".in", "r", stdin), freopen(#a".out", "w", stdout)
#define pv(a) cout << #a << " = " << a << endl
#define pa(a, l, r) cout << #a " : "; rep(_, l, r) cout << a[_] << ' '; cout << endl

const int N = 1e6 + 10;
const int M = 20;

int read() {
  int x = 0, f = 1; char c = getchar();
  for (; c < '0' || c > '9'; c = getchar()) if (c == '-') f = -1;
  for (; c >= '0' && c <= '9'; c = getchar()) x = x * 10 + c - 48;
  return x * f;
}

int n, a[N], p[N], mdep[N], mson[N], dep[N], bfn[N], bn, rev[N], pre[N][M], cnt[N], down[N], lst[N], cur[N], son[N][M], f[N][M];
vector <int> e[N];

signed main() {
  n = read();
  rep(i, 1, n) a[i] = read();
  rep(i, 2, n) {
    p[i] = read();
    // cout << p[i] << ' ' << i << endl;
    e[p[i]].emplace_back(i);
  }
  rep(i, 1, n) mdep[i] = -1;
  drep(i, n, 1) {
    mdep[p[i]] = max(mdep[p[i]], ++ mdep[i]);
    sort(e[i].begin(), e[i].end(), [&](int u, int v) {
      return mdep[u] < mdep[v];
    });
    if (e[i].size()) mson[i] = e[i].back();
  }
  auto bfs = [&](int s = 1) {
    queue <int> q;
    q.push(s);
    while (q.size()) {
      int u = q.front(); q.pop();
      bfn[u] = ++ bn, rev[bfn[u]] = u;
      f[u][0] = f[cur[dep[u]]][0] + a[u];
      cnt[u] = cnt[cur[dep[u]]] + 1;
      rep(i, 0, M - 1) {
        pre[u][i] = pre[cur[dep[u]]][i] + (a[u] >> i & 1);
      }
      lst[u] = cur[dep[u]], cur[dep[u]] = u;
      for (int v : e[u]) {
        dep[v] = dep[u] + 1;
        q.push(v);
      }
    }
  } ;
  bfs();
  rep(u, 1, n) {
    son[u][0] = u;
    cnt[u] += cnt[p[u]];
    rep(i, 0, M - 1) pre[u][i] += pre[p[u]][i];
  }
  drep(u, n, 1) {
    down[u] = down[mson[u]] ? down[mson[u]] : u;
  }
  // pa(p, 1, n);
  // pa(mson, 1, n);
  // pa(lst, 1, n);
  // rep(u, 1, n) cout << f[u][0] << ' ';
  // cout << endl;
  // rep(u, 1, n) cout << pre[u][0] << ' ';
  // cout << endl;
  rep(i, 1, 19) drep(u, n, 1) {
    son[u][i] = son[mson[son[u][i - 1]]][i - 1];
    if (!son[u][i]) continue;
    f[u][i] = f[u][i - 1] + f[mson[son[u][i - 1]]][i - 1];
    int c = pre[son[u][i]][i - 1] - pre[son[u][i - 1]][i - 1];
    f[u][i] += (cnt[son[u][i]] - cnt[son[u][i - 1]] - 2 * c) << i - 1;
    // cout << "i, u, c, f = " << i << ' ' << u << ' ' << c << ' ' << f[u][i] << endl;
    // pv(son[u][i]);
  }
  for (int q = read(); q; -- q) {
    auto ask = [&](int u, int d) {
      if (!u) return 0ll;
      // cout << "----- ask u, d = " << u << ' ' << d << endl;
      d = min(d, dep[down[u]] - dep[u]);
      int res = 0, v = u;
      drep(i, 19, 0) if (d >> i & 1) {
        v = mson[son[v][i]];
      }
      ++ d;
      // pv(v);
      drep(i, 19, 0) if (d >> i) {
        // cout << "u, i, f[u][i] = " << u << ' ' << i << ' ' << f[u][i] << endl;
        res += f[u][i], u = mson[son[u][i]], d ^= 1 << i;
        if (!u || !i || !d) break;
        int c = pre[v][i] - pre[p[u]][i];
        res += (cnt[v] - cnt[p[u]] - 2 * c) << i;
        // cout << "u, i, c, res = " << u << ' ' << i << ' ' << c << ' ' << res << endl;
      }
      return res;
    } ;
    int u = read(), d = read();
    int res = ask(u, d) - ask(lst[u], min(d, dep[down[u]] - dep[u]));
    // pv(ask(u, d));
    // pv(ask(lst[u], d));
    printf("%lld\n", res);
  }
  return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Wrong Answer
time: 2ms
memory: 45468kb

input:

2000
946347867 341162491 202762650 295215762 254064439 66267204 693162580 357612557 492940637 939526638 59775103 374919339 144042807 861369313 651043728 999024805 439554916 167782038 597475252 56704409 69846137 22185655 79439847 769194737 145071391 226046618 915359433 392527325 84982946 54584098 827...

output:

129783432642
558747809
12352523
864526657694
91729843882
585197158
1628586765
922104954
14754466446
158725903988
1832814618
1719999173
57622639685
2418952903
220137077975
165287502244
403506301009
1458062157
139639462122
708763390
974312741
22160481283
541384937
203104970615
755479297
465075002
5120...

result:

wrong answer 1st numbers differ - expected: '31727996563', found: '129783432642'

Test #2:

score: 0
Wrong Answer
time: 131ms
memory: 96100kb

input:

99999
792064428 473106195 799314986 65440734 948726345 442414256 280245405 873012700 466192412 899092866 283555341 657824017 963883713 793944180 767444438 105576842 542107696 580140098 65321660 381184238 584604194 397414881 861590124 309323011 217641157 120832524 303744205 961590116 110259426 380351...

output:

2509771019
102808896745
2506594725
125427731
1983703727
2464509509394
5038222699950
586671581
7983719038922
798898621
468901916
1620113250
25990778
1594379152
8996848302445
30663600697267
1721547881
1491500623
4110192064874
15307884508814
263989598
3068371037
2959123897989
727716410
2234098283207
84...

result:

wrong answer 3rd numbers differ - expected: '528899856', found: '2506594725'

Test #3:

score: 10
Accepted
time: 3688ms
memory: 612884kb

input:

1000000
947727010 179844140 923847675 171881267 5552129 974443359 989307850 869400987 126992154 527448411 141137718 136124474 917813105 392020809 79012342 473860575 969007624 833563354 90169336 878445705 84352622 403307122 733751738 670851448 942399068 731541999 101293644 545785337 964751520 9168003...

output:

322288180595345
64144900444023
131621992874362
185914449206495
4835864536999
52498320047300
91107191167946
133549265294890
287851078359140
123171067016186
128593134440538
8701280311335
14021811618693
65043919486113
310510908841731
15605852792829
75072265994522
73630260235120
66821009550818
217424650...

result:

ok 1000000 numbers

Test #4:

score: 10
Accepted
time: 3703ms
memory: 612852kb

input:

1000000
264862971 751386013 921867736 711577153 262726588 565608444 975324815 440219681 107888226 928241413 729126923 283912914 86248857 896446999 12839598 651796991 139813366 105131395 341646170 839485925 939265720 844548518 102280410 457829889 8602879 737140565 17206920 974175632 535833885 8373832...

output:

1437301063221
97809309869545
338628270305337
233680181746714
7398781587300
271306073103235
20877473367084
90775179215463
2535173725689
66604505519276
211534325021331
84757341592614
17886216279466
132051595671397
1860377236584
46253790496717
83950882453932
257274371637175
240943229927301
118446604805...

result:

ok 1000000 numbers

Test #5:

score: 0
Wrong Answer
time: 1662ms
memory: 597100kb

input:

1000000
978606419 773027389 111265179 979476504 280718851 476857501 751854950 579263969 848569548 781051974 31782627 533831186 812822170 111553645 297770650 331144396 676977983 2236128 258203325 75591120 676466973 60056446 494411414 286185093 92474576 173276071 535648669 87210101 355790411 880267291...

output:

2258826661
938786622
1803221827
34198893150
24126688480
2783437365
16665953904
2720451454
1859550937
476794111
986063591
1579646962
37468743725
38176990
781099105
527479272
5103297773
2519423502
1620138021
45199871146
12472303497
9617012797
2053841976
36282499730
49589751
499629803
9513817887
137931...

result:

wrong answer 4th numbers differ - expected: '20412967786', found: '34198893150'

Test #6:

score: 0
Wrong Answer
time: 2031ms
memory: 597192kb

input:

1000000
952470566 585754087 120174600 401525004 458588768 5487567 31210348 446333263 231409083 521960132 457721893 866842852 925207283 16805978 4706826 99640835 619272676 136536623 459247161 308807462 633687300 717271369 23906473 865522890 173799280 424309108 719410673 118906385 110627845 730629403 ...

output:

994051214
904057442
5097782338
12115414423888
44589763178351
976251948
1386223378
43760580
2742454462
3077329408
127974808928957
716198149
190685967
105215742257612
13827392197052
557364187
1084839575
3104092
276690929
170820739149281
467852673
33049527
425554649
68480583985216
97736610924937
426018...

result:

wrong answer 3rd numbers differ - expected: '2995278795', found: '5097782338'

Test #7:

score: 0
Wrong Answer
time: 1887ms
memory: 597588kb

input:

1000000
732367509 105027907 958920212 886798715 102486738 813075884 301085392 242303497 979657287 944859684 307768 438158233 561755409 740706505 791145209 283862713 828081846 771569552 59044985 600549571 191330226 438693570 36976319 810654215 220068818 771875421 740642902 839964155 206129566 2065543...

output:

999908753
401770479
330074031151912
194002588398182
60975666388213
143172450958334
1146711843
187367763141946
2910590132
4981801495
1079679207
2140524944
75514756
9410084351477
76260056882688
167205592
58363655788956
1085743066
51861395060073
416905260757083
125800058499512
990736484
9627606106
3730...

result:

wrong answer 3rd numbers differ - expected: '206278425790162', found: '330074031151912'

Test #8:

score: 0
Wrong Answer
time: 2239ms
memory: 597424kb

input:

1000000
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...

output:

7054225994
5572189961
25168931
84
9245552261
64321117
0
427455337
125930
1
3
1
1
1
7
0
672988753
1
0
3457875417
8
0
0
17
0
320487913
1
0
0
0
0
853243066
687492110
0
301348
220115086
10
35419106
37297971
3
1
57558122
517977935
3
6772976745
0
1132906086
2
1
4
29406375625
6
7385276870
0
34002718928
121...

result:

wrong answer 1st numbers differ - expected: '3826404725', found: '7054225994'

Test #9:

score: 0
Wrong Answer
time: 2359ms
memory: 597640kb

input:

1000000
465660691 982007525 816592310 377030959 572981469 679249520 86377999 709561525 940473306 35102782 886143915 792819787 903287397 264564177 857982095 91486434 217197704 123118964 383387342 820268798 497623987 255010796 607884194 848568529 38169627 197987657 421323589 664004905 485409127 696844...

output:

96094116015
8243992183131
165758472812666
34208171180799
275400654
1384234654
64716568
11773600395352
10114549437012
263845831954
31172674482652
528398853
1456723425
1800621907
16591607642853
23475037
56966956773988
245789356
109738951
48543618600613
121514115085688
102619945
35215437004742
15560763...

result:

wrong answer 3rd numbers differ - expected: '82893024196511', found: '165758472812666'

Test #10:

score: 0
Wrong Answer
time: 2265ms
memory: 597868kb

input:

1000000
665830082 788228483 245541444 289601309 641764988 150723484 925214020 557415731 310210969 379707835 517820381 883917428 134445288 775557009 444476671 89856268 655841087 888410254 37788122 694551869 563331754 488108584 839551943 415095075 445425438 35452604 562044723 640544531 146258096 66852...

output:

431856043
520373616
139759806500474
7077779085197
101600822883910
13801713959444
762015332
4985151936205
2504048880
594507511
2675566549865
25669163666486
306250439
62500137
90931475805791
786078000
35680002080924
4162775188
4984310933262
877965179
18605465206950
1013123764
8946677188108
56835744100...

result:

wrong answer 3rd numbers differ - expected: '77562434895287', found: '139759806500474'