Submission #2292814
Source Code Expand
#include<bits/stdc++.h> #define maxn 200009 #define LL long long #define pb push_back using namespace std; int a[maxn],sum,rt,f[maxn],sz[maxn],head[maxn],edge=0,fa[maxn]; bool vis[maxn]; LL dis[maxn]; struct da{ int nex,to,w; }e[maxn*2]; struct data{ int x,y;LL w; }; vector<data>E; vector<int>vec; inline void add(int from,int to,int w){ e[++edge]=(da){head[from],to,w}; head[from]=edge; } int find(int x){if(fa[x]!=x)fa[x]=find(fa[x]);return fa[x];} void findrt(int x,int fa){ f[x]=0,sz[x]=1; for(int i=head[x];i;i=e[i].nex){ int v=e[i].to; if(v==fa || vis[v]) continue; findrt(v,x),sz[x]+=sz[v]; f[x]=max(f[x],sz[v]); } f[x]=max(f[x],sum-sz[x]); if(f[x]<f[rt]) rt=x; } void dfs(int x,int fa){ vec.pb(x); for(int i=head[x];i;i=e[i].nex) { int v=e[i].to; if(v==fa || vis[v]) continue; dis[v]=dis[x]+e[i].w; dfs(v,x); } } void solve(int x){ vis[x]=1;vec.clear(); dis[x]=0,dfs(x,0);int p=x; for(int i=0;i<vec.size();i++){ dis[vec[i]]+=a[vec[i]]; if(dis[vec[i]]<dis[p]) p=vec[i]; } for(int i=0;i<vec.size();i++) E.pb((data){vec[i],p,dis[vec[i]]+dis[p]}); for(int i=head[x];i;i=e[i].nex){ int v=e[i].to;if(vis[v]) continue; sum=sz[v],rt=0,findrt(v,x); solve(rt); } } inline bool cmp(const data &A,const data &B){ return A.w<B.w; } int main(){ int n; scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",a+i),fa[i]=i; for(int i=1,x,y,z;i<n;i++) scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z),E.pb((data){x,y,1ll*a[x]+a[y]+z}); sum=n,rt=0,f[0]=1e9,findrt(1,0),solve(1); sort(E.begin(),E.end(),cmp); LL ans=0; for(int i=0;i<E.size();i++){ int u=E[i].x,v=E[i].y; if(find(u)!=find(v)) ans+=E[i].w,fa[find(u)]=find(v); } printf("%lld\n",ans); return 0; }
Submission Info
Submission Time | |
---|---|
Task | J - Tree MST |
User | luogu_bot5 |
Language | C++ (GCC 5.4.1) |
Score | 1900 |
Code Size | 1772 Byte |
Status | AC |
Exec Time | 860 ms |
Memory | 89312 KB |
Compile Error
./Main.cpp: In function ‘int main()’: ./Main.cpp:62:17: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result] scanf("%d",&n); ^ ./Main.cpp:63:48: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result] for(int i=1;i<=n;i++) scanf("%d",a+i),fa[i]=i; ^ ./Main.cpp:65:85: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result] scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z),E.pb((data){x,y,1ll*a[x]+a[y]+z}); ^
Judge Result
Set Name | sample | All | ||||
---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 1900 / 1900 | ||||
Status |
|
|
Set Name | Test Cases |
---|---|
sample | sample-01.txt, sample-02.txt, sample-03.txt |
All | sample-01.txt, sample-02.txt, sample-03.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt, 01-15.txt, 01-16.txt, 01-17.txt, 01-18.txt, 01-19.txt, 01-20.txt, 01-21.txt, 01-22.txt, 01-23.txt, 01-24.txt, 01-25.txt, 01-26.txt, 01-27.txt, 01-28.txt, 01-29.txt, 01-30.txt, 01-31.txt, 01-32.txt, 01-33.txt, 01-34.txt, 01-35.txt, 01-36.txt, 01-37.txt, 01-38.txt, 01-39.txt, 01-40.txt, 01-41.txt, 01-42.txt, 01-43.txt, 01-44.txt, 01-45.txt, 01-46.txt, 01-47.txt, 01-48.txt, 01-49.txt, 01-50.txt, 01-51.txt, 01-52.txt, 01-53.txt, 01-54.txt, 01-55.txt, 01-56.txt, 01-57.txt, 01-58.txt, 01-59.txt, 01-60.txt, 01-61.txt, 01-62.txt, 01-63.txt, 01-64.txt, 01-65.txt, 01-66.txt, 01-67.txt, 01-68.txt, 01-69.txt, 01-70.txt, 01-71.txt, 01-72.txt, 01-73.txt, 01-74.txt, 01-75.txt, 01-76.txt, 01-77.txt, 01-78.txt, 01-79.txt, 01-80.txt, 01-81.txt, 01-82.txt, 01-83.txt, 01-84.txt, 01-85.txt, 01-86.txt, 01-87.txt, 01-88.txt, 01-89.txt, 01-90.txt, sample-01.txt, sample-02.txt, sample-03.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
01-01.txt | AC | 2 ms | 6400 KB |
01-02.txt | AC | 2 ms | 6400 KB |
01-03.txt | AC | 3 ms | 6400 KB |
01-04.txt | AC | 27 ms | 8820 KB |
01-05.txt | AC | 780 ms | 78176 KB |
01-06.txt | AC | 731 ms | 78048 KB |
01-07.txt | AC | 733 ms | 78432 KB |
01-08.txt | AC | 744 ms | 78944 KB |
01-09.txt | AC | 746 ms | 79328 KB |
01-10.txt | AC | 761 ms | 79456 KB |
01-11.txt | AC | 750 ms | 77920 KB |
01-12.txt | AC | 762 ms | 77920 KB |
01-13.txt | AC | 787 ms | 79456 KB |
01-14.txt | AC | 773 ms | 78432 KB |
01-15.txt | AC | 771 ms | 78176 KB |
01-16.txt | AC | 758 ms | 78176 KB |
01-17.txt | AC | 757 ms | 78048 KB |
01-18.txt | AC | 752 ms | 78944 KB |
01-19.txt | AC | 741 ms | 78432 KB |
01-20.txt | AC | 752 ms | 79584 KB |
01-21.txt | AC | 736 ms | 79072 KB |
01-22.txt | AC | 754 ms | 79840 KB |
01-23.txt | AC | 761 ms | 78944 KB |
01-24.txt | AC | 780 ms | 79072 KB |
01-25.txt | AC | 789 ms | 80736 KB |
01-26.txt | AC | 779 ms | 79968 KB |
01-27.txt | AC | 795 ms | 79072 KB |
01-28.txt | AC | 802 ms | 78944 KB |
01-29.txt | AC | 788 ms | 80480 KB |
01-30.txt | AC | 271 ms | 30824 KB |
01-31.txt | AC | 267 ms | 29416 KB |
01-32.txt | AC | 267 ms | 30568 KB |
01-33.txt | AC | 268 ms | 30440 KB |
01-34.txt | AC | 265 ms | 30312 KB |
01-35.txt | AC | 802 ms | 88672 KB |
01-36.txt | AC | 814 ms | 87904 KB |
01-37.txt | AC | 828 ms | 88032 KB |
01-38.txt | AC | 832 ms | 87648 KB |
01-39.txt | AC | 841 ms | 88032 KB |
01-40.txt | AC | 860 ms | 87392 KB |
01-41.txt | AC | 176 ms | 28776 KB |
01-42.txt | AC | 174 ms | 29160 KB |
01-43.txt | AC | 180 ms | 29800 KB |
01-44.txt | AC | 181 ms | 30184 KB |
01-45.txt | AC | 784 ms | 78432 KB |
01-46.txt | AC | 787 ms | 78176 KB |
01-47.txt | AC | 813 ms | 79456 KB |
01-48.txt | AC | 702 ms | 87904 KB |
01-49.txt | AC | 698 ms | 87392 KB |
01-50.txt | AC | 683 ms | 88160 KB |
01-51.txt | AC | 677 ms | 87648 KB |
01-52.txt | AC | 749 ms | 89312 KB |
01-53.txt | AC | 751 ms | 88672 KB |
01-54.txt | AC | 739 ms | 89056 KB |
01-55.txt | AC | 614 ms | 79200 KB |
01-56.txt | AC | 629 ms | 79840 KB |
01-57.txt | AC | 596 ms | 79072 KB |
01-58.txt | AC | 649 ms | 78432 KB |
01-59.txt | AC | 675 ms | 78304 KB |
01-60.txt | AC | 614 ms | 80096 KB |
01-61.txt | AC | 608 ms | 78432 KB |
01-62.txt | AC | 632 ms | 79456 KB |
01-63.txt | AC | 604 ms | 79072 KB |
01-64.txt | AC | 672 ms | 79840 KB |
01-65.txt | AC | 666 ms | 81120 KB |
01-66.txt | AC | 670 ms | 80352 KB |
01-67.txt | AC | 615 ms | 78560 KB |
01-68.txt | AC | 653 ms | 78944 KB |
01-69.txt | AC | 638 ms | 78304 KB |
01-70.txt | AC | 633 ms | 81376 KB |
01-71.txt | AC | 638 ms | 79840 KB |
01-72.txt | AC | 626 ms | 80992 KB |
01-73.txt | AC | 562 ms | 79456 KB |
01-74.txt | AC | 537 ms | 80096 KB |
01-75.txt | AC | 548 ms | 79840 KB |
01-76.txt | AC | 604 ms | 79840 KB |
01-77.txt | AC | 603 ms | 78816 KB |
01-78.txt | AC | 599 ms | 78048 KB |
01-79.txt | AC | 555 ms | 79328 KB |
01-80.txt | AC | 560 ms | 79840 KB |
01-81.txt | AC | 565 ms | 79200 KB |
01-82.txt | AC | 600 ms | 79968 KB |
01-83.txt | AC | 616 ms | 80608 KB |
01-84.txt | AC | 595 ms | 80736 KB |
01-85.txt | AC | 561 ms | 79456 KB |
01-86.txt | AC | 563 ms | 78688 KB |
01-87.txt | AC | 566 ms | 79968 KB |
01-88.txt | AC | 635 ms | 80608 KB |
01-89.txt | AC | 601 ms | 81120 KB |
01-90.txt | AC | 600 ms | 79456 KB |
sample-01.txt | AC | 2 ms | 6400 KB |
sample-02.txt | AC | 2 ms | 6400 KB |
sample-03.txt | AC | 2 ms | 6400 KB |