树分治（挑战p360）

poj1741

题：http://poj.org/problem?id=1741

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=1e18;
int ans=0,tot,maxx,root;
const int M=1e4+4;
struct node{
    int v,w,nextt;
}e[M<<1];
int head[M],vis[M],sz[M],maxv[M],num,dis[M],n,k;
void addedge(int u,int v,int w){
    e[tot].v=v;
    e[tot].w=w;
    e[tot].nextt=head[u];
    head[u]=tot++;
}
void dfssz(int u,int f){
    sz[u]=1;
    maxv[u]=0;
    for(int i=head[u];~i;i=e[i].nextt){
        int v=e[i].v;//cout<<"!!"<<endl;
        if(v==f||vis[v])
            continue;
        dfssz(v,u);
        sz[u]+=sz[v];
        maxv[u]=max(maxv[u],sz[v]);
    }
}
void dfsroot(int r,int u,int f){
    maxv[u]=max(maxv[u],sz[r]-sz[u]);
    if(maxv[u]<maxx){
        maxx=maxv[u];
        root=u;
    }
    for(int i=head[u];~i;i=e[i].nextt){
        int v=e[i].v;
        if(vis[v]||v==f)
            continue;
        dfsroot(r,v,u);
    }
}
void dfsdis(int u,int d,int f){
    dis[num++]=d;
    for(int i=head[u];~i;i=e[i].nextt){
        int v=e[i].v;
        if(v==f||vis[v])
            continue;
        dfsdis(v,d+e[i].w,u);
    }
}
int cal(int u,int d){
    int sum=0;
    num=0;
    dfsdis(u,d,-1);
    sort(dis,dis+num);
    int i=0,j=num-1;
    while(i<j){
        while(dis[i]+dis[j]>k&&i<j){
            j--;
        }
        sum+=j-i;
        i++;
    }
    return sum;
}
void solve(int u){
    maxx=n;
    dfssz(u,-1);
    dfsroot(u,u,-1);
    ans+=cal(root,0);
    
    vis[root]=1;
    for(int i=head[root];~i;i=e[i].nextt){
        int v=e[i].v;
        if(vis[v])
            continue;
        ans-=cal(v,e[i].w);
        solve(v);
    }
}
int main(){
    while(~scanf("%d%d",&n,&k)&&(n+k)){
        tot=0,ans=0;
        memset(head,-1,sizeof(head));
        memset(vis,0,sizeof(vis));
        for(int i=1;i<n;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
            addedge(v,u,w);
        }
        
        solve(1);
        printf("%lld\n",ans);
    }
    return 0;
}

View Code

bzoj 2152

题：https://www.lydsy.com/JudgeOnline/problem.php?id=2152

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int M=2e4+4;
struct node{
    int v,nextt;
    int w;
}e[M<<1];
int head[M],dis[4],sz[M],maxv[M],vis[M],n,tot,fenzi,maxx,root;
void addedge(int u,int v,int w){
    e[tot].v=v;
    e[tot].nextt=head[u];
    e[tot].w=w;
    head[u]=tot++;
}
void dfssz(int u,int f){
    maxv[u]=0;
    sz[u]=1;
    for(int i=head[u];~i;i=e[i].nextt){
        int v=e[i].v;
        if(v==f||vis[v])
            continue;
        dfssz(v,u);
        sz[u]+=sz[v];
        maxv[u]=max(maxv[u],sz[v]);
    }
}
void dfsroot(int r,int u,int f){
    maxv[u]=max(maxv[u],sz[r]-sz[u]);
    if(maxx>maxv[u]){
        maxx=maxv[u];
        root=u;
    }
    for(int i=head[u];~i;i=e[i].nextt){
        int v=e[i].v;
        if(v==f||vis[v])
            continue;
        dfsroot(r,v,u);
    }
}
void dfsdis(int u,int f,int d){
    dis[d%3]++;
    for(int i=head[u];~i;i=e[i].nextt){
        int v=e[i].v;
        if(vis[v]||v==f)
            continue;
        dfsdis(v,u,(d+e[i].w)%3);
    }
}
ll cal(int u,int d){
    dis[0]=dis[1]=dis[2]=0;
    dfsdis(u,-1,d);
    ll sum=0;
    for(int i=0;i<3;i++)
        for(int j=0;j<3;j++)
            if((i+j)%3==0)
                sum+=dis[i]*dis[j];
    return sum;     
}
void solve(int u){
    maxx=n;
    dfssz(u,-1);
    dfsroot(u,u,-1);
    fenzi+=cal(root,0);
    vis[root]=1;
    for(int i=head[root];~i;i=e[i].nextt){
        int v=e[i].v;
        if(vis[v])
            continue;
        fenzi-=cal(v,e[i].w);
        solve(v);
    }
}
int main(){
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    for(int i=1;i<n;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        addedge(u,v,w);
        addedge(v,u,w);
    }
    solve(1);
    int fenmu=n*n;
    //cout<<fenmu<<endl;
    int g=__gcd(fenzi,fenmu);
    printf("%d/%d",fenzi/g,fenmu/g);
    return 0;
}

View Code

 

转载于:https://www.cnblogs.com/starve/p/11562899.html