QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#100962 | #5830. 树 | Scintilla | 0 | 3279ms | 605084kb | C++14 | 3.1kb | 2023-04-28 20:53:44 | 2023-04-28 20:53:46 |
Judging History
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], 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);
mson[p[i]] = i;
}
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;
}
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]);
++ d;
int res = 0;
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) break;
int c = pre[down[u]][i] - pre[p[u]][i];
res += (cnt[down[u]] - 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: 1ms
memory: 37752kb
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:
52316811279 558747809 12352523 10054691833 2759332204 585197158 3183900900 922104954 5892434731 7622694161 1832814618 1719999173 1183105284 2418952875 5872480867 1468711359 21115425148 1458062161 2421020018 708763390 974312741 8298093315 541384937 13651677816 755479297 644887011 51201629 950537853 2...
result:
wrong answer 1st numbers differ - expected: '31727996563', found: '52316811279'
Test #2:
score: 0
Wrong Answer
time: 92ms
memory: 89104kb
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:
1529721251 5789164435 1890036515 125427731 1953894159 15998798166 51167154769 586671581 3349448125 798898621 468901916 1620113250 25990778 1594379152 5312988480 3404424244 1721547881 4111735623 2485379044 3404424244 263989598 1350176621 3406531139 727716410 4142794387 1849681487 5853245640 971328575...
result:
wrong answer 1st numbers differ - expected: '2509771019', found: '1529721251'
Test #3:
score: 0
Wrong Answer
time: 3210ms
memory: 605020kb
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:
322288206991293 64144946606827 131622021461016 185914374898941 4835863616293 52498384514864 91107317145932 133549305860818 287851061406534 123171127466010 128592990936968 8701280183879 14021808260941 65044029324993 310510965687315 15605868051249 75072334423958 73630325382068 66821080528726 217424723...
result:
wrong answer 1st numbers differ - expected: '322288180595345', found: '322288206991293'
Test #4:
score: 0
Wrong Answer
time: 3279ms
memory: 605084kb
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:
1437301279725 97809355735725 338628474888691 233680202642682 7398782854524 271306310490771 20877499166572 90775318291091 2535173933607 66604551363756 211534435306219 84757466685450 17886217839546 132051691526119 1860377260424 46253769112453 83950948773188 257274546429611 240943174832345 118446636358...
result:
wrong answer 1st numbers differ - expected: '1437301063221', found: '1437301279725'
Test #5:
score: 0
Wrong Answer
time: 1488ms
memory: 589340kb
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 1803221825 19115125215 10099675920 3667786822 13439907464 4901567648 1859550937 476794111 986063591 1579646960 8670706325 38176990 781099091 527479272 2392930134 2127007936 1620138021 1586589157 5698332772 9617012771 2053841980 7648109611 49589751 499629803 11482363508 137931191...
result:
wrong answer 3rd numbers differ - expected: '1803221827', found: '1803221825'
Test #6:
score: 0
Wrong Answer
time: 1353ms
memory: 589372kb
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 3574486883 30920858921 5656559259 976251948 369701888 43760580 2742454462 1892746322 5584847963 716198149 190685967 8468000792 1331415448 557364187 1084839579 3104092 276690929 11895002813 467852673 33049531 425554649 3910401634 2133237262 426018723 694554134 676148373 6672719191...
result:
wrong answer 3rd numbers differ - expected: '2995278795', found: '3574486883'
Test #7:
score: 0
Wrong Answer
time: 1353ms
memory: 589768kb
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:
3718354583 401770479 7048649232 30777836909 6534909740 18158881919 1146711843 2020062867 2910590132 4981801489 1079679209 2140524944 75514756 11496763305 27050719749 167205592 5500013057 1085743066 7955775583 268315310 3151106111 990736484 7141737641 1755569740 3147994153 25178730284 829776518 64880...
result:
wrong answer 1st numbers differ - expected: '999908753', found: '3718354583'
Test #8:
score: 0
Wrong Answer
time: 1340ms
memory: 589592kb
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:
244 55 308 -43 2 51 0 132 173 -81 -7 -25 1 1 -3 0 3 3 0 83 -24 0 0 5 0 42 -9 0 0 0 0 117 47 0 45 3 10 11 37 -37 -31 103 24 -3 5 0 82 3 4 10 27 9 12 0 68 287 11 2 -61 -31 -193 1 13 5 0 1 0 2 18 1 6 3 32 0 -14 1 1 8 -13 11 18 180 0 79 -22 0 0 74 0 -72 7 14 0 -12 106 1 3 73 7 0 0 21 1 0 -13 14 324 -29 ...
result:
wrong answer 1st numbers differ - expected: '3826404725', found: '244'
Test #9:
score: 0
Wrong Answer
time: 1370ms
memory: 589780kb
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:
9716240361 7229021080 1638327460 7032230495 275400654 1384234652 64716568 4385262880 8825096713 3824336919 7509509980 528398853 1456723409 1800621907 8971111084 23475037 24196614025 245789356 109738951 4503372779 2023996205 102619945 3058253082 1556076353 5765719027 3001958902 978264700 1182698983 1...
result:
wrong answer 1st numbers differ - expected: '96094116015', found: '9716240361'
Test #10:
score: 0
Wrong Answer
time: 1330ms
memory: 590080kb
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:
431856055 520373616 12101684293 2994711969 3917474960 3036678267 762015338 12334301620 1184478780 594507509 11366404895 1606365176 306250445 62500137 3273984800 786078000 13085512887 4162775188 8001277188 877965179 7212396434 2609908492 10584384696 8791645039 542306751 1978036157 937968345 679655404...
result:
wrong answer 1st numbers differ - expected: '431856043', found: '431856055'