BZOJ 3365: [Usaco2004 Feb]Distance Statistics 路程统计
Posted 北屿
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了BZOJ 3365: [Usaco2004 Feb]Distance Statistics 路程统计相关的知识,希望对你有一定的参考价值。
Description
一棵树,统计距离不大于 \(k\) 的点对个数.
Sol
点分治.
发现自己快把点分治忘干净了...
找重心使所有儿子的最大值尽量小,然后每次处理全部子树,再减去每个子树的贡献,这样就得到子树间的贡献了,然后再搞子树就可以,这就是一个子问题了.
Code
/************************************************************** Problem: 3365 User: BeiYu Language: C++ Result: Accepted Time:352 ms Memory:6128 kb ****************************************************************/ #include<cstdio> #include<utility> #include<vector> #include<queue> #include<algorithm> #include<iostream> using namespace std; typedef pair< int,int > pr; #define mpr make_pair #define debug(a) cout<<#a<<"="<<a<<" " const int N = 40005; int n,m,k,rt,sz,ans; vector<pr> g[N]; int s[N],f[N],b[N]; vector<int> d; int q[N],h,t; inline int in(int x=0,char ch=getchar()){ while(ch>‘9‘ || ch<‘0‘) ch=getchar(); while(ch>=‘0‘ && ch<=‘9‘) x=(x<<3)+(x<<1)+ch-‘0‘,ch=getchar();return x; } void GetRoot(int u,int fa){ s[u]=1,f[u]=0; for(int i=0,v;i<g[u].size();i++) if((v=g[u][i].first)!=fa && !b[v]){ GetRoot(v,u),s[u]+=s[v],f[u]=max(f[u],s[v]); }f[u]=max(f[u],sz-s[u]);if(f[u]<f[rt]) rt=u; } void GetDeep(int u,int fa,int dep){ d.push_back(dep),s[u]=1; for(int i=0,v;i<g[u].size();i++) if((v=g[u][i].first)!=fa && !b[v]) GetDeep(v,u,dep+g[u][i].second),s[u]+=s[v]; } int Calc(int u,int w){ d.clear(),GetDeep(u,u,w); sort(d.begin(),d.end()); int res=0; for(int l=0,r=d.size()-1;l<r;) if(d[l]+d[r]<=k) res+=r-l,l++;else r--; return res; } void GetAns(int u){ ans+=Calc(u,0);b[u]=1; for(int i=0,v;i<g[u].size();i++) if(!b[v=g[u][i].first]) ans-=Calc(v,g[u][i].second),f[0]=sz=s[v],GetRoot(v,rt=0),GetAns(rt); } int main(){ // freopen("in.in","r",stdin); n=in(),m=in(); for(int i=1,u,v,w;i<=m;i++) u=in(),v=in(),w=in(),g[u].push_back(mpr(v,w)),g[v].push_back(mpr(u,w)); k=in(); f[rt=0]=sz=n; GetRoot(1,0); GetAns(rt); cout<<ans<<endl; return 0; }
以上是关于BZOJ 3365: [Usaco2004 Feb]Distance Statistics 路程统计的主要内容,如果未能解决你的问题,请参考以下文章
BZOJ1468: Tree & BZOJ3365: [Usaco2004 Feb]Distance Statistics 路程统计
刷题BZOJ 3365 [Usaco2004 Feb]Distance Statistics 路程统计
bzoj 3365: [Usaco2004 Feb]Distance Statistics 路程统计容斥原理+点分治
bzoj3362/3363/3364/3365[Usaco2004 Feb]树上问题杂烩 并查集/树形dp/LCA/树的点分治