bzoj 3365: [Usaco2004 Feb]Distance Statistics 路程统计【容斥原理+点分治】

统计在一个root下的两个子树，每个子树都和前面的运算一下再加进去对于这种需要排序的运算很麻烦，所以考虑先不去同子树内点对的算出合法点对个数，然后减去每一棵子树内的合法点对（它们实际上是不合法的，相当于一个容斥）
算点对用排序+双指针即可

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1000005;
int n,m,h[N],cnt,rt,sz,si[N],mx[N];
long long dis[N],a[N],tot,ans;
bool v[N];
struct qwe
{int ne,to,va;
}e[N];
int read()
{int r=0,f=1;char p=getchar();while(p>'9'||p<'0'){if(p=='-')f=-1;p=getchar();}while(p>='0'&&p<='9'){r=r*10+p-48;p=getchar();}return r*f;
}
void add(int u,int v,int w)
{cnt++;e[cnt].ne=h[u];e[cnt].to=v;e[cnt].va=w;h[u]=cnt;
}
void gtrt(int u,int fa)
{si[u]=1;mx[u]=0;for(int i=h[u];i;i=e[i].ne)if(e[i].to!=fa&&!v[e[i].to]){gtrt(e[i].to,u);si[u]+=si[e[i].to];mx[u]=max(mx[u],si[e[i].to]);}mx[u]=max(mx[u],sz-si[u]);if(mx[u]<mx[rt])rt=u;
}
void gtde(int u,int fa)
{a[++tot]=dis[u];for(int i=h[u];i;i=e[i].ne)if(e[i].to!=fa&&!v[e[i].to]){dis[e[i].to]=dis[u]+e[i].va;gtde(e[i].to,u);}
}
long long clc(int u)
{tot=0;gtde(u,0);int re=0,l=1,r=tot;sort(a+1,a+tot+1);while(l<r){while(l<r&&a[l]+a[r]>m)   r--;re+=r-l,l++;}return re;
}
void wk(int u)
{v[u]=1;dis[u]=0;ans+=clc(u);for(int i=h[u];i;i=e[i].ne)if(!v[e[i].to]){dis[e[i].to]=e[i].va;ans-=clc(e[i].to);sz=si[e[i].to],rt=0;gtrt(e[i].to,0);wk(rt);}
}
int main()
{n=read(),m=read();for(int i=1;i<n;i++){int x=read(),y=read(),z=read();add(x,y,z),add(y,x,z);}m=read();sz=mx[0]=n;gtrt(1,0);wk(rt);printf("%lld\n",ans);return 0;
}

转载于:https://www.cnblogs.com/lokiii/p/10105488.html

bzoj 3365: [Usaco2004 Feb]Distance Statistics 路程统计【容斥原理+点分治】相关推荐

bzoj 3372: [Usaco2004 Feb]Moo University -- Financial Aid 财政补助（set+贪心）
3372: [Usaco2004 Feb]Moo University -- Financial Aid 财政补助 Time Limit: 10 Sec Memory Limit: 128 MB S ...
bzoj 3367: [Usaco2004 Feb]The Big Game 球赛（DP）
3367: [Usaco2004 Feb]The Big Game 球赛 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 56 Solved: 45 ...
bzoj 3369: [Usaco2004 Feb]Pesky Parentheses 烦人括号（思维题）
3369: [Usaco2004 Feb]Pesky Parentheses 烦人括号 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 14 Sol ...
bzoj 3362: [Usaco2004 Feb]Navigation Nightmare 导航噩梦（加权并查集）
3362: [Usaco2004 Feb]Navigation Nightmare 导航噩梦 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 121 ...
带权并查集【bzoj3362】: [Usaco2004 Feb]Navigation Nightmare 导航噩梦
[bzoj]3362: [Usaco2004 Feb]Navigation Nightmare 导航噩梦农夫约翰有N(2≤N≤40000)个农场,标号1到N,M(2≤M≤40000)条的不同的垂 ...
C++Poisso statistics泊松统计的实现算法(附完整源码)
C++Poisso statistics泊松统计的实现算法 C++Poisso statistics泊松统计的实现算法完整源码(定义,实现,main函数测试) C++Poisso statistics ...
BZOJ_3362_[Usaco2004 Feb]Navigation Nightmare 导航噩梦_并查集
BZOJ_3362_[Usaco2004 Feb]Navigation Nightmare 导航噩梦_并查集 Description 农夫约翰有N(2≤N≤40000)个农场,标号1到N,M( ...
Spark 机器学习概括统计 summary statistics [摘要统计]
概括统计概括统计 summary statistics [摘要统计] 读取要分析的数据,把数据转变成RDD[Vector]类型: 然后,我们调用colStats()方法,得到一个Multivaria ...
BZOJ 1734: [Usaco2005 feb]Aggressive cows 愤怒的牛( 二分答案 )
最小最大...又是经典的二分答案做法.. -------------------------------------------------------------------------- #inc ...

bzoj 3365: [Usaco2004 Feb]Distance Statistics 路程统计【容斥原理+点分治】

bzoj 3365: [Usaco2004 Feb]Distance Statistics 路程统计【容斥原理+点分治】相关推荐

最新文章

热门文章