2011ACM成都赛区现场赛H题 (非递归dfs) (hdu4118)

来源：互联网发布：mac上的空档接龙编辑：程序博客网时间：2024/06/02 01:25

这题我在现场的时候用图论做超时，需要非递归dfs。

我维护了一个a数组记录结点的儿子数。

#include <stdio.h>#include <cstring>#include <stack>using namespace std;struct gtype {       int y,d,next;}g[200010];int first[100100],tot,a[100100],tt,x,y,d,n;bool v[100100];inline int min(int a,int b) {       if (a<b) return a; else return b;} void add(int x,int y,int d) {     tot++;     g[tot].d=d;     g[tot].y=y;     g[tot].next=first[x];     first[x]=tot;     tot++;     g[tot].d=d;     g[tot].y=x;     g[tot].next=first[y];     first[y]=tot;}void dfs(int p) {     stack <int> stk;     int t=first[p];          stk.push(p);     v[p]=true;     while (1) {           if (t==-1) {                      int y = stk.top();                      stk.pop();                      if (stk.empty()) break;                      a[stk.top()]+=a[y];           }           int x = stk.top();           t = first[x];           while (t!=-1) {                 if (!v[g[t].y]) {                                 v[g[t].y]=true;                                 stk.push(g[t].y);                                 break;                 }                 t=g[t].next;           }     }}int main() {    scanf("%d",&tt);    for (int cas=1;cas<=tt;cas++) {        tot=0;        memset(g,0,sizeof(g));        memset(first,-1,sizeof(first));        scanf("%d",&n);        for (int i=1;i<n;i++) {            scanf("%d%d%d",&x,&y,&d);            add(x,y,d);        }        for (int i=1;i<=n;i++) a[i]=1;        memset(v,0,sizeof(v));        dfs(1);        __int64 ans=0;        for (int i=1;i<=2*(n-1);i+=2) {            x=g[i].y;            y=g[i+1].y;            d=min(a[x],a[y]);            ans+=min(d,n-d)*2*g[i].d;        }        printf("Case #%d: %I64d\n",cas,ans);    }    return 0;}