zhxfl hdu 3534 树形dp,求最长路径有多少条,存在负权边.docVIP

#includestdio.h#includestring.h#includeiostream#includevectorusing namespace std;#define maxn 10001#define inf 0x7ffffffint dis[maxn];struct Coor{??? int x,v;};vectorCoorv[maxn];struct Node{??? int Max1,Max2,Max,num1,num2,num;} node[maxn]; //node[x]中记录的是： //x的子节点通过x的最长路径为Max，条数为num， //Max1是x的子节点到x的最长距离，num1是条数 //Max2是x的子节点到x的次长距离，num2是条数 void dfs(int x,int f){??? int i;??? for(i=0; iv[x].size(); i++)??? {??????? int u=v[x][i].x;??????? if(u==f)continue;??????? dfs(u,x);??? }??? dis[x]=node[x].Max1=node[x].Max2=node[x].Max=node[x].num=node[x].num1=node[x].num2=0;??? if(v[x].size()==1x!=1)node[x].num1=node[x].num2=1;//如果是叶子，num1记录为1??? bool flag=1;??? //找出距离最长的叶子??? for(i=0; iv[x].size(); i++)??? {??????? int u=v[x][i].x;??????? int w=v[x][i].v;??????? if(u==f)continue;??????? if(dis[u]+wnode[x].Max1)??????? {??????????? flag=0;??????????? node[x].Max1=dis[u]+w;??????????? node[x].num1=node[u].num1;??????????? node[x].num=0;??????? }??????? else if(dis[u]+w==node[x].Max1)??????? {??????????? node[x].num+=node[x].num1*node[u].num1;//最长的叶子超过1个，可以算出经过x的最长路径有多少个??????????? node[x].num1+=node[u].num1;??????? }??? }??? if(node[x].num==0)//如果最长路径的个数还没有统计出来,统计第二长的??? {??????? for(i=0; iv[x].size(); i++)??????? {??????????? int u=v[x][i].x;??????????? int w=v[x][i].v;??????????? if(u==f)continue;??????????? if(dis[u]+w==node[x].Max1)continue;??????????? if(dis[u]+wnode[x].Max2)??????????? {??????????????? node[x].Max2=dis[u]+w;??????????????? node[x].num2=node[u].num1;??????????? }??????????? else if(dis[u]+w==node[x].Max2)??????????? {??????????????? node[x].num2+=node[u].num1;??????????? }??????? }??????? if(node[x].num2)//第二长的不是0??????? {??????????? node[x].num=node[x].num1*node[x].num2;??????????? node[x].Max=node[x].Max1+node[x].Max2;??????? }??????? else//第二长的为0??????? {??????????? node[x].Max=node[x].Max1;??????????? node[x].num=node[x].num1;??????? }??? }??? else??? {??????? node[x].Max=node