[bzoj 2152] 聪聪可可 [点分治]

Description

聪聪和可可是兄弟俩，他们俩经常为了一些琐事打起来，例如家中只剩下最后一根冰棍而两人都想吃、两个人都想玩儿电脑（可是他们家只有一台电脑）……遇到这种问题，一般情况下石头剪刀布就好了，可是他们已经玩儿腻了这种低智商的游戏。他们的爸爸快被他们的争吵烦死了，所以他发明了一个新游戏：由爸爸在纸上画n个“点”，并用n-1条“边”把这n个“点”恰好连通（其实这就是一棵树）。并且每条“边”上都有一个数。接下来由聪聪和可可分别随即选一个点（当然他们选点时是看不到这棵树的），如果两个点之间所有边上数的和加起来恰好是3的倍数，则判聪聪赢，否则可可赢。聪聪非常爱思考问题，在每次游戏后都会仔细研究这棵树，希望知道对于这张图自己的获胜概率是多少。现请你帮忙求出这个值以验证聪聪的答案是否正确。

Input

输入的第1行包含1个正整数n。后面n-1行，每行3个整数x、y、w，表示x号点和y号点之间有一条边，上面的数是w。

Output

以即约分数形式输出这个概率（即“a/b”的形式，其中a和b必须互质。如果概率为1，输出“1/1”）。

Sample Input

5
1 2 1
1 3 2
1 4 1
2 5 3

Sample Output

13/25

【样例说明】
13组点对分别是(1,1) (2,2) (2,3) (2,5) (3,2) (3,3) (3,4) (3,5) (4,3) (4,4) (5,2) (5,3) (5,5)。

【数据规模】
对于100%的数据，n<=20000。

就当成点分治膜版题贴在这里吧

总有一天要跪着把树形dp的代码也贴上去的

点分治代码：

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<iostream>
  4 #include<vector>
  5 using namespace std;
  6 
  7 inline int read(){
  8     char ch;
  9     int re=0;
 10     bool flag=0;
 11     while((ch=getchar())!='-'&&(ch<'0'||ch>'9'));
 12     ch=='-'?flag=1:re=ch-'0';
 13     while((ch=getchar())>='0'&&ch<='9')  re=re*10+ch-'0';
 14     return flag?-re:re;
 15 }
 16 
 17 struct edge{
 18     int to,w;
 19     edge(int to=0,int w=0):
 20         to(to),w(w){};
 21 };
 22 
 23 const int maxn=20001;
 24 vector<edge> G[maxn];
 25 int n;
 26 int root;
 27 int son[maxn],F[maxn];
 28 bool vis[maxn];
 29 int sum;
 30 int t[3];
 31 int d[maxn];
 32 int ans=0;
 33 
 34 inline void add_edge(int from,int to,int w){
 35     w%=3;
 36     G[from].push_back(edge(to,w));
 37     G[to].push_back(edge(from,w));
 38 }
 39 
 40 void init(){
 41     n=read();
 42     for(int i=0;i<n-1;i++){
 43         int from=read(),to=read(),w=read();
 44         add_edge(from,to,w);
 45     }
 46 }
 47 
 48 void getroot(int x,int fa){
 49     son[x]=1;
 50     F[x]=0;
 51     int dd=G[x].size();
 52     for(int i=0;i<dd;i++){
 53         edge &e=G[x][i];
 54         if(e.to!=fa&&!vis[e.to]){
 55             getroot(e.to,x);
 56             son[x]+=son[e.to];
 57             F[x]=max(F[x],son[e.to]);
 58         }
 59     }
 60     F[x]=max(F[x],sum-son[x]);
 61     if(F[x]<F[root])  root=x;
 62 }
 63 
 64 void get(int x,int fa){
 65     t[d[x]]++;
 66     int dd=G[x].size();
 67     for(int i=0;i<dd;i++){
 68         edge &e=G[x][i];
 69         if(!vis[e.to]&&e.to!=fa){
 70             d[e.to]=(d[x]+e.w)%3;
 71             get(e.to,x);
 72         }
 73     }
 74 }
 75 
 76 int calc(int x,int now){
 77     t[0]=t[1]=t[2]=0;
 78     d[x]=now;
 79     get(x,0);
 80     return t[1]*t[2]*2+t[0]*t[0];
 81 }
 82 
 83 void solve(int x){
 84     ans+=calc(x,0);
 85     vis[x]=1;
 86     int dd=G[x].size();
 87     for(int i=0;i<dd;i++){
 88         edge &e=G[x][i];
 89         if(!vis[e.to]){
 90             ans-=calc(e.to,e.w);
 91             root=0;
 92             sum=son[e.to];
 93             getroot(e.to,0);
 94             solve(root);
 95         }
 96     }
 97 }
 98 
 99 int gcd(int a,int b){
100     return b==0?a:gcd(b,a%b);
101 }
102 
103 int main(){
104     //freopen("temp.in","r",stdin);
105     init();
106     sum=F[root=0]=n;
107     memset(vis,0,sizeof vis);
108     getroot(1,0);
109     solve(root);
110     int t=gcd(ans,n*n);
111     printf("%d/%d
",ans/t,n*n/t);
112     return 0;
113 }

我的理想把我丢在这个拥挤的人潮车窗外已经是一片白雪茫茫