QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#472293#4895. Lovely Dogsczc0 376ms42796kbC++232.1kb2024-07-11 15:30:502024-07-11 15:30:50

Judging History

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

  • [2024-07-11 15:30:50]
  • 评测
  • 测评结果:0
  • 用时:376ms
  • 内存:42796kb
  • [2024-07-11 15:30:50]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
vector<int> G[maxn];
int n,d,a[maxn];
vector<int> D[maxn];//每个数的因子
int g[maxn],prime[maxn],mu[maxn],tot; 
int siz[maxn];
inline void dfs1(int x,int fa){
	if(fa) G[x].erase(find(G[x].begin(),G[x].end(),fa));
	siz[x]=1;
	for(auto &y:G[x]){
		dfs1(y,x);
		siz[x]+=siz[y];
		if(siz[y]>siz[G[x][0]]) swap(y,G[x][0]);
	}
}
int f[maxn],czc[maxn];
int ans;
#define ll long long
const ll INF=1e10;
ll pw[maxn];
inline void add(int x,int s,int op){
	int v=a[x];
	if(op==1){
		for(auto l:D[v]) czc[l]+=g[v]*op;
		for(auto t:D[v]){
			int fl=g[v]*mu[t]*op;
			if(pw[t] && pw[t]/__gcd(pw[t],(ll)v)<=n){
				ans+=fl*czc[pw[t]/__gcd(pw[t],(ll)v)];
			}
		}
	}
	else{
		for(auto t:D[v]){
			int fl=g[v]*mu[t]*op;
			if(pw[t] && pw[t]/__gcd(pw[t],(ll)v)<=n){
				ans+=fl*czc[pw[t]/__gcd(pw[t],(ll)v)];
			}
		}
		for(auto l:D[v]) czc[l]+=g[v]*op;
	}
	for(auto y:G[x]){
		if(y==s) continue;
		add(y,s,op);
	}
}
inline void dfs2(int x,int op){ 
	for(int i=1;i<(int)G[x].size();i++)
		dfs2(G[x][i],0);
	if(G[x].size()) dfs2(G[x][0],1);
	add(x,G[x].size()?G[x][0]:0,1);
	f[x]=ans;
	if(!op){
		add(x,0,-1);
	}
}
int main(){
	scanf("%d%d",&n,&d);
	for(int i=1; ;i++){
		ll v=1;
		for(int j=1;j<=d+1;j++) v*=(ll)i;
		if(v>INF) break;
		pw[i]=v;
	}
	for(int i=1,x,y;i<n;i++){
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=n;i++) for(int j=1;j*i<=n;j++) D[i*j].push_back(i);
	g[1]=mu[1]=1;
	for(int i=2;i<=n;i++){
		if(!prime[i]){
			prime[++tot]=i;
			mu[i]=g[i]=-1;
		}
		for(int j=1;j<=tot && i*prime[j]<=n;j++){
			if(i%prime[j]==0){
				mu[i*prime[j]]=0;
				g[i*prime[j]]=g[i]*g[prime[j]];//完全积性函数的优越性 
				prime[i*prime[j]]=1;
				break;
			}
			else{
				mu[i*prime[j]]=mu[i]*mu[prime[j]];
				g[i*prime[j]]=g[i]*g[prime[j]];
				prime[i*prime[j]]=1;
			}
		}
	}
	dfs1(1,0);
	dfs2(1,1);
	for(int i=1;i<=n;i++) printf("%d\n",f[i]);
	return 0;
}

详细

Subtask #1:

score: 0
Wrong Answer

Test #1:

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

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: 2ms
memory: 10524kb

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:

110
380
9
321
381
122
237
273
118
371
270
117
364
10
384
241
8
272
385
324
380
252
226
382
385
238
238
253
75
371
321
242
422
9
12
323
328
265
0
276
-10
473
8
241
16
452
265
370
276
384
240
325
461
271
386
259
319
320
226
255
282
226
278
251
322
225
372
-3
14
272
8
-1
264
243
471
472
224
265
34
377
...

result:

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

Subtask #3:

score: 0
Wrong Answer

Test #45:

score: 0
Wrong Answer
time: 344ms
memory: 41916kb

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:

143460
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:

wrong answer 1st words differ - expected: '143459', found: '143460'

Subtask #4:

score: 0
Wrong Answer

Test #50:

score: 0
Wrong Answer
time: 371ms
memory: 42796kb

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:

-31724
1
0
0
0
0
0
0
1
0
0
0
-1
0
0
0
0
0
0
1
-1
0
1
0
-1
0
0
1
0
0
-1
0
0
0
0
0
-1
0
0
0
0
0
0
0
0
-1
0
0
0
-1
0
0
0
0
0
0
-1
1
1
0
0
0
-1
0
-1
0
1
0
-1
-1
0
0
0
0
0
0
-1
0
0
0
0
0
-1
0
-1
0
0
0
0
0
0
0
0
1
0
0
0
1
-1
-1
0
0
-1
0
-1
1
-1
0
0
0
-1
0
0
0
0
0
0
0
0
1
-1
0
0
0
1
0
-1
0
0
0
0
0
0
-1
0
0...

result:

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

Subtask #5:

score: 0
Wrong Answer

Test #55:

score: 0
Wrong Answer
time: 376ms
memory: 42700kb

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:

-71573
0
0
0
-1
1
0
-1
-1
0
0
0
0
0
0
0
0
0
0
0
0
-1
-1
0
0
-1
-1
0
-1
0
0
0
0
0
0
0
1
-1
0
0
-1
0
-1
0
0
0
1
-1
0
0
-1
0
0
-1
0
-1
0
1
0
0
0
0
0
0
0
-1
0
0
0
0
0
0
-1
0
-1
0
0
0
-1
0
0
0
0
0
-1
1
0
0
0
0
0
0
1
1
0
0
0
-1
0
0
1
0
1
0
-1
1
-1
0
0
0
0
0
-1
-1
1
0
-1
0
-1
0
0
-1
1
-1
0
1
-1
1
1
1
0
-1
...

result:

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

Subtask #6:

score: 0
Wrong Answer

Test #78:

score: 0
Wrong Answer
time: 185ms
memory: 19660kb

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:

-13293
-261
-1579
-1490
-1054
-2071
91
-745
-1772
-2674
-1351
694
36
-2347
298
-1654
165
-2593
156
-1970
-1045
-1512
852
211
-2021
-2176
-1766
7
-2023
-1767
-1003
-1012
-759
-1574
-2019
-371
-643
382
-2670
16
-2051
-2032
-2289
-270
-636
386
-1522
-679
45
-1594
-1629
-1601
-525
-947
963
-1663
-1526
-...

result:

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

Subtask #7:

score: 0
Time Limit Exceeded

Test #103:

score: 0
Time Limit Exceeded

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:

-130844
-2953
-12693
-8077
-10382
-10884
-1964
624
-10849
-10007
-364
-5602
-2388
-10430
-10635
-10077
-1554
-7696
-7025
-8777
-9845
-10179
-14278
-13527
-2988
1105
-7432
-9154
-10033
-13932
-5858
-8100
933
-289
-10828
-13590
-14130
18
-11008
-9129
-13997
-1834
-3241
-7636
-12106
-12688
-3673
-7953
...

result: