HDU4635(Strongly connected)Tarjan算法,强连通+缩点

 
/* 
 *题目大意： 
 *给你一个DAG图,问你最多能添加多少条边使得这个DAG图依然不是强联通的; 
 * 
 *算法思想： 
 *强连通+缩点 
 *最终添加完边的图,肯定可以分成两个部X和Y,其中只有X到Y的边没有Y到X的边; 
 *那么要使得边数尽可能的多,则X部肯定是一个完全图,Y部也是,同时X部中每个点到Y部的每个点都有一条边; 
 *假设X部有x个点,Y部有y个点,则x+y=n; 
 *同时边数F=x*y+x*(x-1)+y*(y-1),然后去掉已经有了的边m,则为答案; 
 *当x+y为定值时,二者越接近,x*y越大，所以要使得边数最多,那么X部和Y部的点数的个数差距就要越大; 
 *对于给定的有向图缩点,对于缩点后的每个点,如果它的出度或者入度为0,那么它才有可能成为X部或者Y部; 
 *然后找出最大值即可; 
**/  
#include  
#include  
#include  
#include  
using namespace std;  
  
const int N=200010;  
const int M=400010;  
const int INF=0xffffffff;  
  
typedef long long LL;  
  
struct Edge  
{  
    int to,next;  
} edge[M];  
  
LL n,m,cnt,head[N];  
LL dep,top,atype;  
  
LL dfn[N],low[N],vis[N],stack[N],belong[N],in[N],out[N],sum[N];  
  
void addedge(int u,int v)  
{  
    edge[cnt].to=v;  
    edge[cnt].next=head[u];  
    head[u]=cnt++;  
}  
  
void Tarjan(int u)  
{  
    dfn[u]=low[u]=++dep;  
    stack[top++]=u;  
    vis[u]=1;  
    for(int i=head[u]; i!=-1; i=edge[i].next)  
    {  
        int v=edge[i].to;  
        if(!dfn[v])  
        {  
            Tarjan(v);  
            low[u]=min(low[u],low[v]);  
        }  
        else if(vis[v])  
        {  
            low[u]=min(low[u],dfn[v]);  
        }  
    }  
    int j;  
    if(dfn[u]==low[u])  
    {  
        atype++;  
        do  
        {  
            j=stack[--top];  
            belong[j]=atype;  
            sum[atype]++;   //记录每个连通分量中点的个数  
            vis[j]=0;  
        }  
        while(u!=j);  
    }  
}  
  
void solve()  
{  
    if(n==1)  
    {  
        puts("-1");  
        return;  
    }  
    cnt=dep=top=atype=0;  
    memset(head,-1,sizeof(head));  
    memset(dfn,0,sizeof(dfn));  
    memset(low,0,sizeof(low));  
    memset(vis,0,sizeof(vis));  
    memset(belong,0,sizeof(belong));  
    memset(in,0,sizeof(in));  
    memset(out,0,sizeof(out));  
    memset(sum,0,sizeof(sum));  
  
    int u,v;  
    for(int i=0; i