QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#472826 | #4895. Lovely Dogs | liujunyi123 | 0 | 417ms | 57124kb | C++14 | 2.4kb | 2024-07-11 19:39:00 | 2024-07-11 19:39:01 |
Judging History
answer
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=2e5+5;
vector<int> g[N],g1[N];
ll sum,su,f[N],c[N],pw[N];bool t[N];
int n,d,p[N],sz[N],son[N],s[N],a[N],mu[N];
int head[N],cnt;struct node{int to,nxt;}e[N<<1];
void add(int u,int v){e[++cnt]={v,head[u]},head[u]=cnt;}
int ksm(int a,int b){int s=1;
for(;b;b>>=1,a*=a){
if(b&1){if(1ll*s*a>n)return n+1;s*=a;}
if(1ll*a*a>n&&b/2!=0)return n+1;
}
return s;
}
void init(int u,int fa){sz[u]=1;
for(int i=head[u],v;i;i=e[i].nxt)if((v=e[i].to)!=fa){
init(v,u),sz[u]+=sz[v];
if(sz[v]>sz[son[u]])son[u]=v;
}
}
void insert(int u){ll tot=0;
tot+=su;int i=0;
for(auto x:g[p[u]]){
int y=p[u],cnt=d+1-g1[p[u]][i];
tot+=mu[x]*c[ksm(x,cnt)];
// cout<<" "<<u<<" "<<mu[x]<<" "<<c[ksm(x,cnt)]<<" "<<ksm(x,cnt)<<" "<<x<<" "<<cnt<<endl;
}
sum+=a[u]*tot,su+=a[u];
// cout<<sum<<" "<<su<<endl;
for(auto x:g[p[u]])c[x]+=a[u];//cout<<" "<<u<<" "<<x<<" "<<a[u]<<endl;;
}
void del(int u){su=0;for(auto x:g[p[u]])c[x]=0;}
void ins(int u,int fa){insert(u);for(int i=head[u],v;i;i=e[i].nxt)if((v=e[i].to)!=fa)ins(v,u);}
void clr(int u,int fa){del(u);for(int i=head[u],v;i;i=e[i].nxt)if((v=e[i].to)!=fa)clr(v,u);}
void dfs(int u,int fa){
for(int i=head[u],v;i;i=e[i].nxt)if((v=e[i].to)!=fa&&v!=son[u])dfs(v,u),s[u]+=s[v],sum=su=0,clr(v,u);
if(son[u])dfs(son[u],u),s[u]+=s[son[u]];
for(int i=head[u],v;i;i=e[i].nxt)if((v=e[i].to)!=fa&&v!=son[u])ins(v,u);
// cout<<endl;
insert(u),f[u]=sum+s[u];
// cout<<u<<" "<<sum<<" "<<s[u]<<" "<<p[u]<<endl;
// for(int i=1;i<=n;i++)cout<<c[i]<<" ";
// cout<<endl;
}
int main(){
scanf("%d%d",&n,&d);
for(int i=1,x,y;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x);
for(int i=1;i<=n;i++)scanf("%d",&p[i]),pw[i]=ksm(i,d),t[pw[i]]=1;
for(int i=1;i<=n;i++){mu[i]+=i==1;for(int j=i+i;j<=n;j+=i)mu[j]-=mu[i];}
for(int i=2;i<=n;i++)for(int j=i;j<=n;j+=i)g[j].push_back(i);
for(int i=1;i<=n;i++){s[i]=a[i]=1;
long long y=1ll*p[i]*p[i],z=p[i];
for(auto x:g[p[i]]){int cnt=0,ct1=0;
while(y%x==0)cnt++,y/=x;
while(z%x==0)ct1++,z/=x;
if(cnt>d)s[i]=0;else s[i]*=((cnt&1)?-1:1);
if(ct1>d)a[i]=0;else a[i]*=((ct1&1)?-1:1);
}
for(auto x:g[p[i]]){int cnt=0,y=p[i];
while(y%x==0)cnt++,y/=x;
g1[p[i]].push_back(cnt);
}
// cout<<a[i]<<" "<<s[i]<<endl;
}
init(1,0),dfs(1,0);
for(int i=1;i<=n;i++)printf("%d\n",f[i]);//cout<<i<<endl;
return 0;
}
詳細信息
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 0
Wrong Answer
time: 0ms
memory: 19272kb
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:
15 1 1 1 0 1 0 11 3 1 6 1 3 1 2 1 1 6 1 0
result:
wrong answer 1st words differ - expected: '16', found: '15'
Subtask #2:
score: 0
Wrong Answer
Test #24:
score: 10
Accepted
time: 5ms
memory: 19256kb
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:
581 -3 0 0 0 0 0 0 0 -2 0 0 0 0 0 -1 0 0 0 1 0 -1 0 0 -1 0 0 0 17 -2 0 -1 -2 0 0 0 0 0 0 0 -5 0 0 0 0 -14 0 -1 0 -1 0 0 1 1 -1 -4 0 0 1 0 0 0 3 0 0 0 -1 -2 0 0 4 0 0 0 0 -1 0 1 0 0 0 -5 0 0 0 0 -1 0 0 0 0 0 0 0 1 -1 0 18 0 0 13 -2 0 -2 0 0 0 0 2 -2 2 0 0 3 0 -1 0 0 0 0 -3 0 0 0 0 0 0 1 -1 0 0 0 0 0 ...
result:
ok 2000 tokens
Test #25:
score: 0
Accepted
time: 5ms
memory: 20128kb
input:
2000 1 1754 1650 906 642 596 1542 1656 1549 716 1578 1799 1182 53 244 1032 41 1290 1758 485 1496 1438 948 1683 684 400 653 1756 1459 1965 1322 1540 1263 1365 1564 108 1801 741 717 1113 13 1787 1124 411 732 64 1817 907 259 1308 29 1518 752 375 422 663 1631 528 799 863 310 790 793 587 579 1828 874 502...
output:
581 0 -1 -1 0 0 0 0 0 0 0 0 0 0 0 0 0 -1 0 0 -12 0 0 0 0 0 0 2 -1 0 0 0 0 0 0 0 0 0 -1 -1 1 0 0 -2 0 20 -1 0 -3 1 0 5 -1 0 0 1 1 0 0 0 0 -1 0 -2 0 0 0 -15 1 1 0 0 0 0 0 1 0 30 0 1 0 0 1 -1 0 0 0 0 -1 0 0 0 0 0 1 0 0 0 0 0 0 0 -1 0 1 0 1 -1 0 -1 0 0 0 0 -3 0 0 0 0 0 0 0 0 0 0 0 0 0 0 -1 0 0 0 1 0 0 0...
result:
ok 2000 tokens
Test #26:
score: 0
Accepted
time: 0ms
memory: 22528kb
input:
2000 1 146 1160 146 388 146 1033 382 1917 162 1342 1 1425 1841 764 1674 780 1109 1649 1282 1786 488 1386 1753 1698 17 192 1692 944 693 146 1933 146 976 463 1603 392 1709 248 18 678 146 1157 1517 1416 31 1153 973 39 1359 1046 625 1840 745 146 1316 146 124 146 627 1410 146 540 772 1461 1041 1537 1374 ...
output:
581 0 -36 -192 435 473 506 0 0 358 0 0 0 0 0 0 -180 35 438 0 0 0 607 0 0 0 -17 0 0 26 2 499 0 -180 -85 0 -104 -120 -63 -60 0 0 0 0 598 0 0 0 0 356 592 0 0 0 45 -73 -116 343 0 0 0 0 0 0 31 -108 -5 0 0 0 637 270 494 0 0 487 0 -197 0 401 520 0 0 52 0 128 0 0 82 0 0 0 -22 0 0 0 0 0 0 30 0 0 0 0 0 0 0 0 ...
result:
ok 2000 tokens
Test #27:
score: 0
Accepted
time: 0ms
memory: 19092kb
input:
2000 1 681 278 1551 1142 424 928 738 174 1393 1727 456 944 1713 468 359 1597 1265 1737 246 500 1095 695 654 904 1465 27 1172 1385 1455 40 1391 1384 1979 970 1123 800 1618 1892 1444 1506 79 806 313 1350 1872 85 1467 1031 741 1139 739 1681 263 1454 169 885 1222 153 864 799 192 1339 935 1843 1633 1358 ...
output:
581 0 -1 0 0 0 1 25 0 0 0 0 1 0 -8 0 0 -3 0 0 0 -2 0 0 -3 0 0 0 0 0 0 0 -3 -1 -1 0 0 -1 1 -2 3 0 0 0 0 0 2 3 0 0 -1 -7 0 0 0 0 0 -7 0 0 -1 1 9 0 0 -1 0 0 0 0 0 -1 -3 -1 1 -3 0 -1 -1 0 -1 -1 -1 -1 0 0 -2 -1 12 -7 -10 0 0 0 -6 0 0 0 0 41 0 0 -15 0 0 0 0 0 1 1 0 0 0 0 0 -7 0 -3 -26 0 0 0 0 0 0 0 0 0 2 ...
result:
ok 2000 tokens
Test #28:
score: -10
Wrong Answer
time: 2ms
memory: 19356kb
input:
2000 2 1608 842 1808 1921 1404 549 594 1521 1755 855 1047 1256 340 1877 407 670 1100 1239 1511 1142 790 1103 1212 944 515 167 180 415 399 1563 1458 136 728 1480 1074 819 555 1594 1693 1301 1802 1879 1936 501 306 87 1125 796 720 1298 1999 1529 767 1396 1258 1940 1651 1564 1059 281 704 848 1861 473 13...
output:
1916 3 2 12 1 1 1 1 49 1 1 1 2 0 4 3 7 1 -1 1 0 1 1 1 1 0 4 5 0 0 1 119 0 2 12 45 0 0 0 1 1 1 1 1 1 1 1 7 1 0 -2 1 1 3 0 0 0 0 2 0 1 0 3 1 1 5 1 -2 1 1 2 1 2 0 0 6 0 1 1 1 6 1 1 1 0 0 3 1 0 1 1 1 4 1 1 0 1 4 0 -1 1 1 0 1 1 1 3 0 0 0 1 1 0 1 1 1 0 0 2 1 24 1 0 1 1 0 1 1 2 0 1 4 1 3 0 2 1 1 0 3 3 2 1 ...
result:
wrong answer 1st words differ - expected: '2854', found: '1916'
Subtask #3:
score: 0
Wrong Answer
Test #45:
score: 0
Wrong Answer
time: 417ms
memory: 57124kb
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:
143448 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: '143448'
Subtask #4:
score: 0
Time Limit Exceeded
Test #50:
score: 0
Time Limit Exceeded
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:
result:
Subtask #5:
score: 0
Time Limit Exceeded
Test #55:
score: 0
Time Limit Exceeded
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:
result:
Subtask #6:
score: 0
Time Limit Exceeded
Test #78:
score: 0
Time Limit Exceeded
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:
result:
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...