1. 程式人生 > >HDU 6445 Search for Answer(最小費用最大流-mcmf)

HDU 6445 Search for Answer(最小費用最大流-mcmf)

Description

給出一個nn個點的完全圖的鄰接矩陣aa,其中ai,j=1a_{i,j}=1表示i,ji,j之間邊的方向是iijjai,j=0a_{i,j}=0表示i,ji,j之間邊的方向是jjiiai,j=aj,i=2a_{i,j}=a_{j,i}=2表示i,ji,j之間邊的方向尚未確定,要求給不確定方向的邊定向使得對該矩陣執行下面程式碼所得結果最大

img

Input

第一行一整數TT表示用例組數,每組用例首先輸入一整數nn表示點數,之後輸入一個n×nn\times n的矩陣表示該圖的鄰接矩陣,其中a

i,j=aj,i=2a_{i,j}=a_{j,i}=2的位置至多200200

(2T10,5n200)(2\le T\le 10,5\le n\le 200)

Output

輸出定向後所給程式碼執行結果最大值

Sample Input

2 5 02112 20221 02001 02102 20020 5 01112 00022 01012 02002 22220

Sample Output

40 24

Solution

對於每個四元團a,b,c,da,b,c,d,有三種情況:

1.四點成環,也即沒有出度為22的點,對答案貢獻為11

2.有兩個點的出度為22,對答案貢獻為

1-1

3.只有一個點的出度為11,對答案貢獻為00

假設所有四元團貢獻均為11,這樣的貢獻為An4A_n^4,之後考慮負貢獻,如果一個四元團只有一個出度為22的點則負貢獻11,如果一個四元團有兩個出度為22的點則負貢獻為22,故只要對每個點累加考慮其作為22出度點時的情況數即為所有負貢獻

假設沒有不確定的邊,考慮每個點對答案的貢獻,對於點aa,假設其出度為du(a)du(a),從其出度點中任選兩個點b,db,d的方案數為Adu(a)2A_{du(a)}^2,從剩餘n3n-3個點中任選一個點cc即可,注意到選出的這四個點只是給出了相對順序,而絕對順序(也就是迴圈中的先後順序)有四種,進而有 a

ns=An4a=1n4(n3)Adu(a)2=An48(n3)a=1nCdu(a)2 ans=A_n^4-\sum\limits_{a=1}^n4\cdot (n-3)\cdot A_{du(a)}^2=A_n^4-8\cdot (n-3)\cdot \sum\limits_{a=1}^nC_{du(a)}^2 故我們需要給邊定向使得a=1nCdu(a)2\sum\limits_{a=1}^nC_{du(a)}^2最小,對於點aa,設與其相連的邊中有xx條不確定方向的邊,aa已確定的出度依舊記為du(a)du(a),如果這xx條邊中有yy條確定從aa出發,那麼aa對答案的貢獻即為 Cdu(a)+y2=Cdu(a)2+i=0y1(du(a)+i) C_{du(a)+y}^2=C_{du(a)}^2+\sum\limits_{i=0}^{y-1}(du(a)+i) 如果用yy個流的限制來表示aa會多出yy個出度,從aa向匯點建xx條邊,每條邊流量為11,費用依次為du(a),du(a)+1,...,du(a)+x1du(a),du(a)+1,...,du(a)+x-1,那麼滿流下的最小費用即為aa在保證多出yy個出度時對答案的最小貢獻

進一步的,考慮所有點對答案的最小貢獻和,一排點表示沒有確定方向的邊,一排點表示這nn個點,每條邊向其兩個端點建流量為11費用為00的邊,源點向每條邊建流量為11費用為00的邊,每個點向匯點建邊如前,那麼該網路的最大流即為給每條邊確定方向,最小費用即為確定方向後所有點對答案的貢獻和最小值

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
#define maxn 405
#define maxm 100005
#define INF 0x3f3f3f3f
int head[maxn],d[maxn],s,e,no,vis[maxn],pre[maxn];//s為源點,e為匯點 
struct point
{
    int u,v,flow,next,cost;
    point(){};
    point(int _u,int _v,int _next,int _flow,int _cost)
	{
		u=_u,v=_v,next=_next,flow=_flow,cost=_cost;
	} 
}p[maxm];
void add(int x,int y,int z,int c)//從x到y建一條容量為z,花費為c的邊 
{
    p[no]=point(x,y,head[x],z,c);   
    head[x]=no++;
    p[no]=point(y,x,head[y],0,-c);  
    head[y]=no++;
}
void init()//初始化 
{
    memset(head,-1,sizeof(head));
    no=0;
}
bool spfa()
{
    int i,x,y;
    queue<int>q;
    memset(d,0x3f,sizeof(d));
    memset(vis,false,sizeof(vis));
    memset(pre,-1,sizeof(pre));
    d[s]=0;   
    vis[s]=true;  
    q.push(s);
    while(!q.empty())
    {
        x=q.front();  
        q.pop();    
        vis[x]=false;
        for(i=head[x];i!=-1;i=p[i].next)
        {
            if(p[i].flow&&d[y=p[i].v]>d[x]+p[i].cost)
            {
                d[y]=d[x]+p[i].cost;   
                pre[y]=i;
                if(vis[y])  
                    continue;
                vis[y]=true;  
                q.push(y);
            }  
        }  
    }
    return d[e]!=d[e+1];
}
int mcmf()//最小費用最大流 
{
    int mincost=0,maxflow=0,minflow,i;
    while(spfa())
    {
    	//if(d[e]>=0)break;//可行流 
        minflow=INF;
        for(i=pre[e];i!=-1;i=pre[p[i].u])
            minflow=min(minflow,p[i].flow);
        for(i=pre[e];i!=-1;i=pre[p[i].u])
        {
            p[i].flow-=minflow;
            p[i^1].flow+=minflow;
        }
        mincost+=d[e]*minflow; 
        maxflow+=minflow;
    }
    return mincost;//最小費用,最大流為maxflow 
}
int T,n,m,num[205],du[205];
char c[205];
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)num[i]=du[i]=0;
		init();
		m=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%s",c+1);
			for(int j=1;j<=n;j++)
			{
				int x=c[j]-'0';
				if(x==1)du[i]++;
				else if(x==2&&i<j)
				{
					m++,num[i]++,num[j]++;
					add(n+m,i,1,0);
					add(n+m,j,1,0);
				}
			}
		}
		s=0,e=n+m+1;
		for(int i=1;i<=m;i++)add(s,n+i,1,0);
		int res=0;
		for(int i=1;i<=n;i++)
		{
			res+=du[i]*(du[i]-1)/2;
			for(int j=0;j<num[i];j++)add(i,e,1,du[i]++);
		}
		res=8*(n-3)*(res+mcmf());
		printf("%d\n",n*(n-1)*(n-2)*(n-3)-res);
	}
	return 0;
}