BZOJ 2435 道路修建 NOI2011 树形DP

　　一看到这道题觉得很水，打了递归树形DP后RE了一组，后来发现必须非递归(BFS)

递归版本84分：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int N,point[],next[],v[],c[],cnt=,f[];
bool p[];
long long sum=;
void insect(int x,int y,int z){next[cnt]=point[x];point[x]=cnt;v[cnt]=y;c[cnt]=z;cnt++;}
void dfs(int x)
{
    int i;
    f[x]=;
    for (i=point[x];i!=-;i=next[i])
     if (p[v[i]]==)
         {
             p[v[i]]=;
            dfs(v[i]);
            f[x]+=f[v[i]];
            sum+=(1LL*c[i]*abs(*f[v[i]]-N));
        }
}
int main()
{
    int i,x,y,z;
    memset(point,-,sizeof(point));
    memset(next,-,sizeof(next));
    memset(v,,sizeof(v));
    memset(p,,sizeof(p));
    memset(c,,sizeof(c));
    memset(f,,sizeof());
    scanf("%d\n",&N);
    for (i=;i<N;++i)
    {
        scanf("%d %d %d\n",&x,&y,&z);
        insect(x,y,z); insect(y,x,z);
    }p[]=;dfs();
    printf("%lld\n",sum);
    return ;
}

BFS版本AC：

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int N,point[],next[],v[],c[],cnt=,f[],s[],fa[];
int table[],num=,du[];
bool p[];
long long sum=;
void insect(int x,int y,int z){next[cnt]=point[x];point[x]=cnt;v[cnt]=y;c[cnt]=z;cnt++;}
void dfs(int x)
{
    int now,i,tmp=;
    queue<int>q;
    for (i=;i<=N;++i)
    {
        if (du[i]==)
         q.push(i),f[i]=;
        f[i]=;
    }
    while (!q.empty())
    {
        now=q.front(); q.pop(); p[now]=; tmp=now;
        for (i=point[now];i!=-;i=next[i])
         if (p[v[i]]==)
         {
             f[v[i]]+=f[now];
             du[v[i]]--;
             if (du[v[i]]==)
             q.push(v[i]);
         }
    }
    memset(p,,sizeof(p));
    p[tmp]=; q.push(tmp);
    while (!q.empty())
    {
        now=q.front(); q.pop();
        for (i=point[now];i!=-;i=next[i])
         if (p[v[i]]==)
         {
             p[v[i]]=;
             sum+=(1LL*c[i]*abs(*f[v[i]]-N));
             q.push(v[i]);
         }
    }
    printf("%lld\n",sum);
}
int main()
{
    int i,x,y,z;
    memset(point,-,sizeof(point));
    memset(next,-,sizeof(next));
    memset(du,,sizeof(du));
    memset(v,,sizeof(v));
    memset(p,,sizeof(p));
    memset(c,,sizeof(c));
    memset(f,,sizeof());
    scanf("%d\n",&N);
    for (i=;i<N;++i)
    {
        scanf("%d %d %d\n",&x,&y,&z); du[x]++; du[y]++;
        insect(x,y,z); insect(y,x,z);
    }dfs();
    return ;
}