Submission #2292802
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 | zjo_2001 |
Language | C++14 (GCC 5.4.1) |
Score | 1900 |
Code Size | 1847 Byte |
Status | AC |
Exec Time | 854 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 | 3 ms | 6396 KB |
01-02.txt | AC | 2 ms | 6400 KB |
01-03.txt | AC | 2 ms | 6400 KB |
01-04.txt | AC | 27 ms | 8820 KB |
01-05.txt | AC | 773 ms | 79840 KB |
01-06.txt | AC | 745 ms | 78304 KB |
01-07.txt | AC | 747 ms | 79456 KB |
01-08.txt | AC | 739 ms | 79328 KB |
01-09.txt | AC | 743 ms | 78816 KB |
01-10.txt | AC | 755 ms | 79584 KB |
01-11.txt | AC | 756 ms | 79200 KB |
01-12.txt | AC | 763 ms | 78944 KB |
01-13.txt | AC | 767 ms | 79200 KB |
01-14.txt | AC | 781 ms | 79456 KB |
01-15.txt | AC | 783 ms | 78432 KB |
01-16.txt | AC | 764 ms | 78304 KB |
01-17.txt | AC | 764 ms | 79968 KB |
01-18.txt | AC | 753 ms | 78688 KB |
01-19.txt | AC | 731 ms | 78432 KB |
01-20.txt | AC | 750 ms | 79200 KB |
01-21.txt | AC | 743 ms | 79968 KB |
01-22.txt | AC | 749 ms | 79072 KB |
01-23.txt | AC | 761 ms | 79200 KB |
01-24.txt | AC | 780 ms | 79328 KB |
01-25.txt | AC | 786 ms | 79200 KB |
01-26.txt | AC | 795 ms | 80736 KB |
01-27.txt | AC | 790 ms | 79456 KB |
01-28.txt | AC | 806 ms | 80864 KB |
01-29.txt | AC | 786 ms | 79072 KB |
01-30.txt | AC | 269 ms | 30696 KB |
01-31.txt | AC | 270 ms | 29032 KB |
01-32.txt | AC | 261 ms | 29288 KB |
01-33.txt | AC | 256 ms | 29160 KB |
01-34.txt | AC | 253 ms | 28904 KB |
01-35.txt | AC | 800 ms | 88288 KB |
01-36.txt | AC | 817 ms | 89184 KB |
01-37.txt | AC | 825 ms | 88544 KB |
01-38.txt | AC | 830 ms | 88928 KB |
01-39.txt | AC | 843 ms | 88928 KB |
01-40.txt | AC | 854 ms | 87776 KB |
01-41.txt | AC | 176 ms | 30056 KB |
01-42.txt | AC | 175 ms | 29672 KB |
01-43.txt | AC | 176 ms | 30184 KB |
01-44.txt | AC | 175 ms | 29928 KB |
01-45.txt | AC | 784 ms | 78688 KB |
01-46.txt | AC | 785 ms | 79200 KB |
01-47.txt | AC | 806 ms | 78816 KB |
01-48.txt | AC | 690 ms | 87264 KB |
01-49.txt | AC | 700 ms | 88032 KB |
01-50.txt | AC | 682 ms | 88800 KB |
01-51.txt | AC | 673 ms | 88160 KB |
01-52.txt | AC | 737 ms | 88800 KB |
01-53.txt | AC | 739 ms | 89312 KB |
01-54.txt | AC | 732 ms | 89184 KB |
01-55.txt | AC | 615 ms | 78432 KB |
01-56.txt | AC | 627 ms | 79968 KB |
01-57.txt | AC | 589 ms | 78304 KB |
01-58.txt | AC | 645 ms | 78688 KB |
01-59.txt | AC | 680 ms | 78560 KB |
01-60.txt | AC | 608 ms | 78304 KB |
01-61.txt | AC | 609 ms | 79456 KB |
01-62.txt | AC | 624 ms | 79328 KB |
01-63.txt | AC | 600 ms | 79584 KB |
01-64.txt | AC | 685 ms | 80864 KB |
01-65.txt | AC | 684 ms | 80992 KB |
01-66.txt | AC | 679 ms | 79584 KB |
01-67.txt | AC | 602 ms | 79328 KB |
01-68.txt | AC | 634 ms | 78432 KB |
01-69.txt | AC | 630 ms | 78560 KB |
01-70.txt | AC | 628 ms | 80480 KB |
01-71.txt | AC | 636 ms | 81632 KB |
01-72.txt | AC | 626 ms | 81248 KB |
01-73.txt | AC | 560 ms | 78688 KB |
01-74.txt | AC | 535 ms | 78304 KB |
01-75.txt | AC | 545 ms | 78944 KB |
01-76.txt | AC | 604 ms | 78560 KB |
01-77.txt | AC | 606 ms | 79328 KB |
01-78.txt | AC | 602 ms | 78560 KB |
01-79.txt | AC | 555 ms | 80096 KB |
01-80.txt | AC | 557 ms | 78944 KB |
01-81.txt | AC | 567 ms | 78944 KB |
01-82.txt | AC | 600 ms | 80092 KB |
01-83.txt | AC | 618 ms | 80864 KB |
01-84.txt | AC | 594 ms | 80352 KB |
01-85.txt | AC | 560 ms | 79712 KB |
01-86.txt | AC | 561 ms | 79072 KB |
01-87.txt | AC | 567 ms | 80096 KB |
01-88.txt | AC | 635 ms | 81888 KB |
01-89.txt | AC | 600 ms | 80480 KB |
01-90.txt | AC | 603 ms | 81248 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 |