[bzoj2152]聰聰可可(點分治)
阿新 • • 發佈:2017-09-09
return calc con urn type || rac getch main
題意:在一棵樹中,求兩點之間距離是3的倍數的概率。
解題關鍵:$p = \frac{{\sum {nu{m_0}*nu{m_0} + nu{m_1}*nu{m_2}*2} }}{{n*n}}$
求概率分子分母必須對應,分母是無序的,則分子也必須是無序的。
復雜度:$O(n\log n)$
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #include<cstdlib> 5 #include<iostream> 6 #include<cmath> 7#define inf 0x3f3f3f3f 8 #define maxn 20004 9 using namespace std; 10 typedef long long ll; 11 int head[maxn],cnt,n,k=3,ans,size,s[maxn],f[maxn],root,depth[maxn],num;//vis代表整體的訪問情況,每個dfs不應該只用vis來存儲 12 bool vis[maxn]; 13 ll ans1,ans2,ans3; 14 struct edge{ 15 int to,w,nxt; 16 }e[maxn<<1]; 17 void add_edge(int u,int v,int w){ 18 e[cnt].to=v; 19 e[cnt].w=w; 20 e[cnt].nxt=head[u]; 21 head[u]=cnt++; 22 } 23 24 inline int read(){ 25 char k=0;char ls;ls=getchar();for(;ls<‘0‘||ls>‘9‘;k=ls,ls=getchar()); 26 int x=0;for(;ls>=‘0‘&&ls<=‘9‘;ls=getchar())x=(x<<3)+(x<<1)+ls-‘0‘; 27 if(k==‘-‘)x=0-x;return x; 28 } 29 30 void get_root(int u,int fa){//get_root會用到size 31 s[u]=1;f[u]=0;//f是dp數組 32 for(int i=head[u];i!=-1;i=e[i].nxt){ 33 int v=e[i].to; 34 if(v==fa||vis[v]) continue; 35 get_root(v,u); 36 s[u]+=s[v]; 37 f[u]=max(f[u],s[v]); 38 } 39 f[u]=max(f[u],size-s[u]); 40 root=f[root]>f[u]?u:root; 41 } 42 43 void get_depth_size(int u,int fa,int dis){//同時獲取size和depth 44 if(dis%3==0) ans1++; 45 else if(dis%3==1) ans2++; 46 else ans3++; 47 s[u]=1; 48 for(int i=head[u];i!=-1;i=e[i].nxt){ 49 int v=e[i].to; 50 if(v==fa||vis[v]) continue; 51 get_depth_size(v,u,dis+e[i].w); 52 s[u]+=s[v]; 53 } 54 } 55 56 int calc(int u,int fa,int w){ 57 ans1=ans2=ans3=0; 58 get_depth_size(u,fa,w); 59 return ans1*ans1+ans2*ans3*2; 60 } 61 62 void work(int u){ 63 vis[u]=true; 64 ans+=calc(u,-1,0); 65 for(int i=head[u];i!=-1;i=e[i].nxt){ 66 int v=e[i].to; 67 if(vis[v]) continue; 68 ans-=calc(v,u,e[i].w); 69 size=s[v],root=0; 70 get_root(v,u); 71 work(root); 72 } 73 } 74 75 void init(){ 76 memset(vis,false, sizeof vis); 77 memset(head,-1,sizeof head); 78 ans=cnt=0; 79 } 80 81 82 int main(){ 83 int a,b,c; 84 f[0]=inf; 85 while(scanf("%d",&n)!=EOF){ 86 init(); 87 for(int i=0;i<n-1;i++){ 88 a=read(),b=read(),c=read(); 89 add_edge(a,b,c); 90 add_edge(b,a,c); 91 } 92 size=n,root=0; 93 get_root(1,-1); 94 work(root); 95 int gg=__gcd(ans,n*n); 96 printf("%d/%d\n",ans/gg,n*n/gg); 97 } 98 return 0; 99 100 }
[bzoj2152]聰聰可可(點分治)