bzoj 2599: [IOI2011]Race

 依次处理子树，用t数组表示取了x价值的最小深度，这样就可以避免每次重复计算一棵子树的情况，每次统计的答案就是min{d[x]+t[K-x]}

 1 #include<bits/stdc++.h>
 2 #define N 100005
 3 #define M 10000005
 4 #define LL long long
 5 #define inf 0x3f3f3f3f
 6 using namespace std;
 7 inline int ra()
 8 {
 9     int x=0,f=1; char ch=getchar();
10     while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
11     while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
12     return x*f;
13 }
14 int n,K,cnt,sum,root,ans;
15 int t[1000005],head[200005],size[200005],f[200005],dis[200005],d[200005];
16 bool vis[200005];
17 struct edge{int to,next,v;}e[400005];
18 void insert(int x, int y, int v){e[++cnt].next=head[x]; e[cnt].to=y; e[cnt].v=v; head[x]=cnt;}
19 void getroot(int x, int fa)
20 {
21     size[x]=1; f[x]=0;
22     for (int i=head[x];i;i=e[i].next)
23     {
24         if (e[i].to==fa || vis[e[i].to]) continue;
25         getroot(e[i].to,x);
26         size[x]+=size[e[i].to];
27         f[x]=max(f[x],size[e[i].to]);
28     }
29     f[x]=max(f[x],sum-size[x]);
30     if (f[x]<f[root]) root=x;
31 }
32 void cal(int x, int fa)
33 {
34     if (dis[x]<=K) ans=min(ans,d[x]+t[K-dis[x]]);
35     for (int i=head[x];i;i=e[i].next)
36     {
37         if (e[i].to==fa || vis[e[i].to]) continue;
38         dis[e[i].to]=dis[x]+e[i].v;
39         d[e[i].to]=d[x]+1;
40         cal(e[i].to,x);
41     }
42 }
43 void add(int x, int fa, int flag)
44 {
45     if (dis[x]<=K)
46     {
47         if (flag) t[dis[x]]=min(t[dis[x]],d[x]);
48         else t[dis[x]]=n;
49     }
50     for (int i=head[x];i;i=e[i].next)
51         if (e[i].to!=fa && !vis[e[i].to])
52             add(e[i].to,x,flag);
53 }
54 void work(int x)
55 {
56     vis[x]=1; t[0]=0;
57     for (int i=head[x];i;i=e[i].next)
58     {
59         if (vis[e[i].to]) continue;
60         d[e[i].to]=1; dis[e[i].to]=e[i].v;
61         cal(e[i].to,0);
62         add(e[i].to,0,1);
63     }
64     for (int i=head[x];i;i=e[i].next)
65         if (!vis[e[i].to]) add(e[i].to,0,0);
66     for (int i=head[x];i;i=e[i].next)
67     {
68         if (vis[e[i].to]) continue;
69         root=0; sum=size[e[i].to];
70         getroot(e[i].to,0);
71         work(root);
72     }
73 }
74 int main()
75 {
76     n=ra(); K=ra();
77     for (int i=1; i<=K; i++) t[i]=n;
78     for (int i=1; i<n; i++)
79     {
80         int x=ra()+1,y=ra()+1,v=ra();
81         insert(x,y,v); insert(y,x,v);
82     }
83     ans=sum=f[0]=n;
84     getroot(1,0);
85     work(root);
86     if (ans!=n) printf("%d\n",ans); else printf("-1\n");
87     return 0;
88 }

 

posted @ 2017-02-20 13:56  ws_ccd  阅读(158)  评论(0)  编辑  收藏  举报