QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#740440#4895. Lovely DogsNineSuns10 439ms62420kbC++142.5kb2024-11-13 09:56:302024-11-13 09:56:31

Judging History

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

  • [2024-11-13 09:56:31]
  • 评测
  • 测评结果:10
  • 用时:439ms
  • 内存:62420kb
  • [2024-11-13 09:56:30]
  • 提交

answer

#include <bits/stdc++.h>
#define ll long long
#define pii pair <int, int>
#define fi first
#define se second
#define pb push_back

using namespace std;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
const int N = 2e5+5; 
int n, d, v[N], pr[N], pc, sz[N], gs[N], a[N];
ll f[N], ans[N], cs[N]; 
vector <int> e[N], tx, dv[N], td[N];  

inline int F (int x) {
	int res = 1; 
	while (x > 1) {
		int k = v[x], c = 0; 
		while (x%k == 0) {
			++c; res = -res; x /= k; 
		}
	}
	return res; 
}

void init (int n) {
	for (int i = 2;i <= n;i++) {
		if (!v[i]) { v[i] = i; pr[++pc] = i; }
		for (int j = 1;j <= pc && pr[j]*i <= n;j++) {
			v[pr[j]*i] = pr[j]; 
			if (pr[j] == v[i]) break; 
		}
	}
	for (int i = 1;i <= n;i++) for (int j = i;j <= n;j += i) dv[j].pb(i); 
	for (int i = 1;i <= n;i++) {
		int x = i; 
		while (x > 1) {
			int k = v[x], c = 0; 
			while (x%k == 0) {
				++c; x /= k; 
			}
			ll v = 1; 
			int q = max(d+1-c, 0); 
			while (q--) {
				v = min(1ll*n+1, v*k); 
			}
			td[i].pb(v); 
		}
	}
}


void dfs (int p, int fa) {
	sz[p] = 1; 
	for (int i : e[p]) {
		if (i == fa) continue;
		dfs(i, p); sz[p] += sz[i]; 
		if (sz[i] > sz[gs[p]]) gs[p] = i; 
	}
}

ll res; 
void del (int p, int fa) {
	for (int j : dv[a[p]]) cs[j] -= f[a[p]]; 
	for (int i : e[p]) {
		if (i == fa) continue; 
		del(i, p); 
	}
}
void ad (int k) {
	for (int j : dv[k]) cs[j] += f[k]; 
	for (int j = 0;j < (1<<td[k].size());j++) {
		ll v = 1; 
		for (int z = 0;z < td[k].size();z++) {
			if (j>>z&1^1) continue; 
			v = min(1ll*n+1, v*td[k][z]); 
		}
		if (__builtin_popcount(j)&1) res -= f[k]*cs[v]; else res += f[k]*cs[v]; 
	}
}

void add (int p, int fa) {
	ad(a[p]); 
	for (int i : e[p]) {
		if (i == fa) continue; 
		add(i, p); 
	}
}

void dfs2 (int p, int fa) {
	for (int i : e[p]) {
		if (i == fa || i == gs[p]) continue; 
		dfs2(i, p); del(i, p); res = 0; 
	}
	if (gs[p]) dfs2(gs[p], p); 
	ad(a[p]); 
	for (int i : e[p]) {
		if (i == fa || i == gs[p]) continue;
		add(i, p);  
	}
	ans[p] = res; 
}

void solve () {
	cin >> n >> d;
	init(n); 
	for (int i = 1;i <= n;i++) f[i] = F(i); 
	for (int i = 1;i < n;i++) {
		int x, y; cin >> x >> y; 
		e[x].pb(y); e[y].pb(x); 
	}
	for (int i = 1;i <= n;i++) cin >> a[i]; 
	dfs(1, 0);
	dfs2(1, 0); 
	for (int i = 1;i <= n;i++) cout << ans[i] << "\n"; 
}

signed main () {
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	solve(); 
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 0
Wrong Answer
time: 5ms
memory: 22812kb

input:

20 2
18 8
18 11
13 19
10 8
9 11
4 8
9 15
9 17
2 1
13 18
20 18
1 8
12 17
7 16
5 11
16 15
6 19
14 16
1 3
2 15 5 13 20 6 16 18 9 19 17 7 14 10 11 3 1 12 4 8

output:

14
1
1
1
0
1
0
10
2
1
4
1
3
1
0
0
1
5
1
0

result:

wrong answer 1st words differ - expected: '16', found: '14'

Subtask #2:

score: 0
Wrong Answer

Test #24:

score: 0
Wrong Answer
time: 6ms
memory: 22996kb

input:

2000 1
134 1468
867 1750
351 1220
1690 1888
1685 134
585 282
1142 643
206 271
260 1833
1987 770
1029 1667
322 1371
341 518
601 915
119 893
1933 1502
951 1785
1056 1630
1957 1208
96 55
1508 1212
331 427
505 151
1378 1486
1545 697
1459 629
202 997
180 1917
1638 1177
1244 1896
302 658
1433 1605
1318 19...

output:

-329
-2
0
0
0
0
0
0
0
-2
0
0
0
0
0
-1
0
0
0
1
0
-2
0
0
0
0
0
0
28
-2
0
-1
-2
0
0
0
0
0
0
0
-7
0
0
0
0
-24
0
-2
0
-1
-12
0
-5
-1
-1
4
0
0
2
0
0
0
3
0
0
0
-2
-2
0
0
7
0
0
0
-2
-1
0
1
0
0
0
-1
0
0
0
0
-1
0
0
0
0
0
-1
-1
-2
-1
0
31
0
0
12
-2
0
5
0
0
0
0
3
-2
4
0
0
6
0
-1
0
0
0
0
-3
0
0
0
0
0
-2
1
0
0
0
...

result:

wrong answer 1st words differ - expected: '581', found: '-329'

Subtask #3:

score: 10
Accepted

Test #45:

score: 10
Accepted
time: 329ms
memory: 53796kb

input:

200000 20
117994 12616
53490 106425
103660 50033
132640 78252
58384 19939
69183 10015
39098 165030
179856 130356
65245 57831
18234 83378
4240 154896
177149 102260
4634 180087
132390 19627
98506 60775
1890 120740
87908 21917
41323 192721
181885 96684
69412 139951
9800 38301
59025 29879
186185 81402
1...

output:

143459
1
6
1
17
19
1
2
3
1
1
26
1
1
1
1
1
1
7
1
1
3
1
2
1
1
1
1
4
3
2
1
3
13
19
1
1
1
1
1
1
34
1
1
1
12
1
1
3
1
24
1
1
1
2
1
1
6
1
3
1
3
1
10
1
1
1
3
4
9
1
2
3
1
1
1
7
2
6
1
1
1
1
1
1
2
3
457
2
1
3
1
1
1
1
12
1
1
1
6
2
1
1
3
1
3
113
1
4
11
2
1
4
2
1
1
1
25
22
2
3
1
1
3
3
1
2
3
1
2
1
1
107
1
1
14
1
1...

result:

ok 200000 tokens

Test #46:

score: 10
Accepted
time: 358ms
memory: 53876kb

input:

200000 20
26219 163867
20331 153212
126612 40599
53814 68996
79701 140933
144374 181902
59705 155221
11230 70725
158998 133605
163268 88141
83507 114091
7736 162046
143360 92662
197974 194981
129770 126237
133117 75376
44213 67464
131083 19290
35473 65770
192299 66427
112908 181240
139699 88439
1103...

output:

143459
1
1
3
3
1
4
1
1
1
17
2
4
2
1
1
1
1
2
1
1
1
2
1
1
1
9
1
2
2
1
1
1
1
2
1
1
3
1
2
7
1
1
1
1
1
3
6
14
20
1
1
1
7
7
1
52
1
6
1
1
24
5
3
3
1
1
1
1
1
12
1
1
16
1
1
4
1
1
1
1
9
1
1
1
1
1
5
1
1
120
1
22
1
1
1
4
15
1
3
3
1
1
1
1
48
1
1
1
1
1
1
13
1
3
3
1
1
7
1
5
3
1
1
1
1
1
1
1
783
3
1
1
1
1
2
1
1
1
1
...

result:

ok 200000 tokens

Test #47:

score: 10
Accepted
time: 217ms
memory: 62420kb

input:

200000 20
151436 178769
79211 141557
185103 181923
173473 84222
84222 31293
159673 17537
72335 131147
59289 141263
84222 23691
163353 84222
76461 84222
80651 84307
76526 12294
199619 84222
59457 177309
19912 1273
164867 44284
41990 26506
40399 84222
98264 145475
84222 48507
84222 88935
84222 176452
...

output:

143459
94518
71442
82904
129477
42163
72542
1
1
45757
1
130757
52193
1
1
1
96408
1
78846
1
57773
1
1
1
86255
119507
1
1
83415
1
1
1
1
1
84437
1
1
1
72508
1
77107
84355
1
1
1
72616
52116
1
136452
1
100206
131528
1
146256
76408
1
1
1
1
50456
1
1
1
1
75137
1
83751
71005
116031
1
52173
1
75120
1
80850
1...

result:

ok 200000 tokens

Test #48:

score: 10
Accepted
time: 439ms
memory: 53616kb

input:

200000 20
76510 76490
88933 21393
126948 187403
137672 130527
82789 167591
134447 15851
54831 5084
196062 114272
151180 77255
51713 92637
179118 81158
109526 64703
34747 40350
96352 50618
67033 44700
33353 157246
193080 130434
169961 20611
11637 109101
191766 55895
98648 132015
126097 100752
187559 ...

output:

143459
1
2
28
1
2
2
1
1
6
1
4
6
2
4
1
2
1
1
1
6
6
72
1
1
1
1
8
1
2
188
2
2
1
4
2
6
4
76
6
6
6
8
2
1
2
28
1
8
1
2
1
6
1
1862
1
1
1
20
28
64
1
1
4
1
1
2
2
1
6
4
1
1
1
1
20
1
1
1
1
8
1
8
1
1
2
4
1
1
48
12
2
1
2
4
6
1
1
8
2
2
1
1
1
1
1
2
1
8
1
1
1
1
1
8
6
2
32
1
1
2
36
1
2
8
1
1
56
1
1
1
1
2
1
1
1
2
2
2...

result:

ok 200000 tokens

Test #49:

score: 10
Accepted
time: 356ms
memory: 53888kb

input:

200000 20
91828 79744
98337 148526
56405 54432
49447 185733
136527 109193
20160 117165
95938 77898
169046 98403
45584 26426
55328 117425
132115 35999
174350 136071
10629 151193
127424 169510
172287 15111
72777 59970
110820 119084
188317 89900
195364 7345
77707 106548
139440 165710
26624 117460
11202...

output:

143459
16
1
1
1
1
1
1
1
1
2
1
2
9
2
1
1
1
1
1
1
1
4
2
2
1
2
1
1
6
1
8
1
1
1
1
1
1
11
2
1
1
1
1
1
1
37
1
3
1
7
6
2
1
1
1
1
1
10
2
1
6
1
15
1
11
66
3
1
1
1
12
3
5
2
1
1
15
1
7
2
7
12
3
1
1
29
3
1
128
1
2
1
1
3
11
1
2
1
3
1
1
1
1
1
1
1
13
1
1
7
1
7
69
1
2
2
1
14
1
1
1
16
1
1
1
1
1
2
2
1
1
1
1
3
42
2
9
...

result:

ok 200000 tokens

Subtask #4:

score: 0
Wrong Answer

Test #50:

score: 0
Wrong Answer
time: 178ms
memory: 53656kb

input:

200000 1
1 2
1 3
1 4
1 5
1 6
1 7
1 8
1 9
1 10
1 11
1 12
1 13
1 14
1 15
1 16
1 17
1 18
1 19
1 20
1 21
1 22
1 23
1 24
1 25
1 26
1 27
1 28
1 29
1 30
1 31
1 32
1 33
1 34
1 35
1 36
1 37
1 38
1 39
1 40
1 41
1 42
1 43
1 44
1 45
1 46
1 47
1 48
1 49
1 50
1 51
1 52
1 53
1 54
1 55
1 56
1 57
1 58
1 59
1 60
1 61...

output:

-31677
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0...

result:

wrong answer 1st words differ - expected: '-44916', found: '-31677'

Subtask #5:

score: 0
Wrong Answer

Test #55:

score: 0
Wrong Answer
time: 166ms
memory: 53100kb

input:

200000 1
1 2
1 3
1 4
1 5
1 6
1 7
1 8
1 9
1 10
1 11
1 12
1 13
1 14
1 15
1 16
1 17
1 18
1 19
1 20
1 21
1 22
1 23
1 24
1 25
1 26
1 27
1 28
1 29
1 30
1 31
1 32
1 33
1 34
1 35
1 36
1 37
1 38
1 39
1 40
1 41
1 42
1 43
1 44
1 45
1 46
1 47
1 48
1 49
1 50
1 51
1 52
1 53
1 54
1 55
1 56
1 57
1 58
1 59
1 60
1 61...

output:

-71526
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0...

result:

wrong answer 1st words differ - expected: '-44916', found: '-71526'

Subtask #6:

score: 0
Wrong Answer

Test #78:

score: 0
Wrong Answer
time: 55ms
memory: 30488kb

input:

50000 1
8097 41839
17674 41774
40520 8024
5786 38261
20664 43471
1217 49276
11185 40807
14186 25584
31704 14814
42333 41475
13053 39565
45938 30104
5826 39463
5031 10814
43784 6042
58 33849
42978 18978
36307 33276
34769 4351
27884 37532
27528 29431
29451 39345
10946 9667
19016 47269
7911 30103
10308...

output:

-10569
0
0
0
0
0
0
-1
0
-3
0
0
1
0
0
0
0
0
0
0
0
-2
-1
-1
0
-2
0
2
-1
4
0
0
-1
0
0
0
0
0
0
0
-1
-2
-1
0
-1
0
-3
0
2
0
0
3
-1
2
0
0
24
0
0
0
0
0
0
-6
0
-2
30
-2
12
-1
1
5
0
0
1
0
-1
0
0
1
4
-2
-1
0
-1
0
0
0
0
0
-1
0
0
-2
0
-1
0
0
0
0
0
0
0
0
0
2
-17
0
0
0
0
0
0
0
8
0
1
0
0
0
0
0
0
0
0
0
0
0
0
0
1
0
0...

result:

wrong answer 1st words differ - expected: '-9152', found: '-10569'

Subtask #7:

score: 0
Wrong Answer

Test #103:

score: 0
Wrong Answer
time: 332ms
memory: 53956kb

input:

200000 1
118863 188865
188022 168616
118976 119404
178852 33449
81624 40431
151228 160976
68943 136313
57200 117631
147789 139875
100240 55537
164811 145415
103548 186750
15010 168029
155731 107005
69836 1502
86171 122700
83448 131948
189162 94464
128210 2509
49724 183329
174782 192641
27687 71315
1...

output:

-108256
0
0
0
0
0
0
0
0
1
-1
-1
0
0
0
0
0
0
-1
-1
0
0
-1
0
0
0
0
0
0
-1
0
0
0
0
0
0
0
0
0
10
0
0
-1
0
0
0
0
0
0
18
0
0
0
-1
0
0
-3
0
1
6
-20
0
1
0
0
0
0
0
1
0
0
0
3
0
0
-1
-22
0
0
0
0
0
0
-3
0
0
0
0
0
0
0
0
0
0
0
0
-69
0
0
0
-1
-1
0
0
0
0
-1
-1
0
-3
0
13
0
0
0
0
2
0
-1
0
0
0
3
0
0
0
0
0
-2
0
0
-2
0
...

result:

wrong answer 1st words differ - expected: '-44916', found: '-108256'