QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#100965 | #5830. 树 | Scintilla | 20 | 3623ms | 605100kb | C++14 | 3.3kb | 2023-04-28 21:14:10 | 2023-04-28 21:14:11 |
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]) son[u][i] = son[lst[u]][i];
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;
// if (u == 5) 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: 7ms
memory: 40552kb
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 2418952903 5872480867 1468711361 21115425148 1458062157 2421020018 708763390 974312741 8298093315 541384937 13651677816 755479297 644887011 51201629 950537857 2...
result:
wrong answer 1st numbers differ - expected: '31727996563', found: '52316811279'
Test #2:
score: 0
Wrong Answer
time: 90ms
memory: 89596kb
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 1849681495 5853245640 971328575...
result:
wrong answer 1st numbers differ - expected: '2509771019', found: '1529721251'
Test #3:
score: 10
Accepted
time: 3613ms
memory: 605100kb
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: 3623ms
memory: 605028kb
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: 1444ms
memory: 589260kb
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 19115125215 10099675920 3667786828 13439907464 4901567648 1859550937 476794111 986063591 1579646962 8670706325 38176990 781099105 527479272 2392930136 2127007940 1620138021 1586589157 5698332772 9617012797 2053841976 7648109611 49589751 499629803 11482363508 137931191...
result:
wrong answer 4th numbers differ - expected: '20412967786', found: '19115125215'
Test #6:
score: 0
Wrong Answer
time: 1475ms
memory: 589440kb
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 5656559251 976251948 369701890 43760580 2742454462 1892746322 5584847963 716198149 190685967 8468000792 1331415448 557364187 1084839575 3104092 276690929 11895002813 467852673 33049527 425554649 3910401634 2133237262 426018723 694554152 676148373 6672719191...
result:
wrong answer 3rd numbers differ - expected: '2995278795', found: '3574486883'
Test #7:
score: 0
Wrong Answer
time: 1403ms
memory: 589728kb
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 2020062869 2910590132 4981801495 1079679207 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: 1257ms
memory: 589652kb
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 13 2 51 0 132 173 1 3 1 1 1 1 0 11 1 0 83 8 0 0 3 0 42 1 0 0 0 0 117 47 0 45 3 10 11 37 1 1 103 24 3 5 0 82 3 4 4 27 9 12 0 68 287 3 2 1 3 15 1 13 3 0 1 0 2 18 1 6 3 32 0 6 1 1 16 1 3 18 180 0 79 2 0 0 74 0 2 7 14 0 2 106 1 1 73 7 0 0 21 1 0 1 14 324 1 1 6 300 239 10 14 21 7 165 1 4 2 0 8...
result:
wrong answer 1st numbers differ - expected: '3826404725', found: '244'
Test #9:
score: 0
Wrong Answer
time: 1266ms
memory: 589784kb
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 1384234654 64716568 4385262880 8825096713 3824336919 7509509980 528398853 1456723425 1800621907 8971111084 23475037 24196614025 245789356 109738951 4503372779 2023996205 102619945 3058253082 1556076353 5765719027 3001958902 978264700 1182698995 1...
result:
wrong answer 1st numbers differ - expected: '96094116015', found: '9716240361'
Test #10:
score: 0
Wrong Answer
time: 1292ms
memory: 590068kb
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 12101684293 2994711969 3917474960 3036678265 762015332 12334301620 1184478776 594507511 11366404895 1606365176 306250439 62500137 3273984800 786078000 13085512887 4162775188 8001277188 877965179 7212396434 2609908492 10584384696 8791645039 542306751 1978036157 937968345 679655404...
result:
wrong answer 3rd numbers differ - expected: '77562434895287', found: '12101684293'