UVALive-6528 Disjoint water supply

// 题意是给你一个DAG// 让你找到有不同路径的点对数量// 不同路径定义为除了根节点1以外，其他的路径上的点都不同// 那么首先我们考虑树型DAG// 那么怎么样的点对有不同的路径呢// 只要在不同的子树上就一定有不同的路径，因为他们的最近公共祖先是根1// 而在同一颗子树上的点对一定会有公共点是子树的根// 所以只需要有每个节点对应的是那颗子树就好了，标记一个fa[x]就好了// 那么现在考虑一般的DAG，假设已经知道了按照拓扑序算出了所在点一定会经过那个点//，标记为fa[x]// 那么现在一个节点有多个前驱节点，也就是父节点，那么如果父节点的fa[x]都相同，// 那么该点也一定会经过fa[x],所以就可以标记他的fa 为 fa[x] ，// 而如果他的前驱节点有不同的fa[x],那么该点与任意一个点都一定有不同的路径咯，// 所以就可以标记一个新的fa[q]=q// 然后后面的点如此处理即可//////　给一从1出发的DAG，问有多少个点对到点1存在两条互不相交的路径。 //　对于每一个记录一下fa[x]。 //　1.如果x与1相连，fa[x]=x //　2.如果x的前驱结点fa都相同就将其赋为fa[x]=该相同值，不然fa[x]=x //　3.fa相同的点在一组，那么相交。////===#include<cstdio>#include<cstring>#include<algorithm>#include<iostream>using namespace std;const int maxn=1000+10;const int maxm=100000+10;//struct Edge{//  int u,v,next;//  Edge(){}//  Edge(int u,int v,int next):u(u),v(v),next(next){}//};//Edge edges[maxm*2];//bool gd[maxn][maxn];int C,P;int fa[maxn];int topo[maxn];int vis[maxn];int cnt=0;void init(){    cnt=0;    memset(vis,0,sizeof(vis));    memset(topo,0,sizeof(topo));}void dfs(int u){    for(int i=u+1;i<=C;i++){        if(gd[u][i]){            if(!vis[i]){                vis[i]=1;                dfs(i);            }        }    }    topo[cnt++]=u;}void solve(){    int u,v;    memset(gd,0,sizeof(gd));    for(int i=1;i<=P;i++){        scanf("%d %d",&u,&v);        gd[u][v]=1;    }    //queue<int> que;    //memset(fa,0,sizeof(fa));    //for(int i=2;i<=C;i++){    //  if(gd[1][i]){    //      fa[i]=i;    //      que.push(i);    //  }    //}    //while(!que.empty()){    //  u=que.front();que.pop();    //  for(int i=u+1;i<=C;i++){    //    //}    init();    vis[1]=1;    dfs(1);    memset(fa,0,sizeof(fa));    fa[1]=1;    for(int i=2;i<=C;i++)        if(gd[1][i])            fa[i]=i;    for(int i=cnt-1;i>=0;i--){        if(fa[topo[i]]) continue;        bool flag=1;        int pre=0;        for(int j=1;j<topo[i];j++){            if(gd[j][topo[i]]==0)   continue;            if(pre&&pre!=fa[j]){                flag=0;                break;            }            else pre=fa[j];        }        if(flag)    fa[topo[i]]=pre;        else fa[topo[i]]=topo[i];    }//  for(int i=1;i<=C;i++){//        printf("%d\n",fa[i]);//    }    int num[maxn];    memset(num,0,sizeof(num));    for(int i=1;i<=C;i++){        num[fa[i]]++;    }    int res=0;    for(int i=1;i<=C;i++){        for(int j=i+1;j<=C;j++){        res+=num[i]*num[j];        }    }    printf("%d\n",res);}int main(){    while(scanf("%d %d",&C,&P)==2){        solve();    }    return 0;}