BZOJ 1827 洛谷 2986 [USACO10MAR]伟大的奶牛聚集Great Cow Gather

Posted Driver_Lao

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了BZOJ 1827 洛谷 2986 [USACO10MAR]伟大的奶牛聚集Great Cow Gather相关的知识,希望对你有一定的参考价值。

【题解】

  很容易想到暴力做法,枚举每个点,然后对于每个点O(N)遍历整棵树计算答案。这样整个效率是O(N^2)的,显然不行。

  我们考虑如果已知当前某个点的答案,如何快速计算它的儿子的答案。

  显然选择它的儿子作为集合点,它的儿子的子树内的奶牛可以少走当前点到儿子节点的距离dis,不在它儿子的子树内的奶牛要多走dis. 那么我们维护每个节点的子树内的奶牛总数(即点权和),就可以快速进行计算了。效率O(N).

  

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define N 200010
 4 #define rg register
 5 #define LL long long
 6 using namespace std;
 7 int n,tot,last[N],v[N];
 8 LL ans,sum,size[N],dis[N];
 9 struct edge{
10     int to,pre,dis;
11 }e[N];
12 inline int read(){
13     int k=0,f=1; char c=getchar();
14     while(c<\'0\'||c>\'9\')c==\'-\'&&(f=-1),c=getchar();
15     while(\'0\'<=c&&c<=\'9\')k=k*10+c-\'0\',c=getchar();
16     return k*f;
17 }
18 void dfs(int x,int fa){
19     size[x]=v[x];
20     for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa){
21         dis[to]=dis[x]+e[i].dis; 
22         dfs(to,x); size[x]+=size[to];
23     }
24 }
25 void solve(int x,int fa,LL now){
26     ans=min(ans,now);
27     for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa){
28         LL tmp=now-size[to]*e[i].dis+(sum-size[to])*e[i].dis;
29         solve(to,x,tmp);
30     }
31 }
32 int main(){
33     n=read(); 
34     for(rg int i=1;i<=n;i++) v[i]=read(),sum+=v[i];
35     for(rg int i=1;i<n;i++){
36         int u=read(),v=read(),w=read();
37         e[++tot]=(edge){v,last[u],w}; last[u]=tot;
38         e[++tot]=(edge){u,last[v],w}; last[v]=tot;
39     }
40     dfs(1,0);
41     for(rg int i=1;i<=n;i++) ans+=dis[i]*v[i];
42     solve(1,0,ans);
43     printf("%lld\\n",ans);
44     return 0;
45 }
View Code

 

以上是关于BZOJ 1827 洛谷 2986 [USACO10MAR]伟大的奶牛聚集Great Cow Gather的主要内容,如果未能解决你的问题,请参考以下文章

洛谷 P2986 [USACO10MAR]伟大的奶牛聚集(树形动规)

洛谷 P2986 [USACO10MAR]伟大的奶牛聚集Great Cow Gat…

BZOJ 2986 莫比乌斯函数+容斥原理

洛谷P1827 美国血统 American Heritage

洛谷 P1827 美国血统 American Heritage

bzoj 2986 Non-Squarefree Numbers 容斥原理+数学