[HNOI2014]画框
题目描述
小T准备在家里摆放几幅画,为此他买来了N幅画和N个画框。为了体现他的品味,小T希望能合理地搭配画与画框,使得其显得既不过于平庸也不太违和。
对于第 幅画与第 个画框的配对,小T都给出了这个配对的平凡度Aij 与违和度Bij 。整个搭配方案的总体不和谐度为每对画与画框平凡度之和与每对画与画框违和度的乘积。具体来说,设搭配方案中第i幅画与第Pi个画框配对,则总体不和谐度为 小T希望知道通过搭配能得到的最小的总体不和谐度是多少。
输入输出格式
输入格式:输入文件第 行是一个正整数T ,表示数据组数,接下来是T组数据。对于每组数据,第 行是一个正整数N,表示有N对画和画框。第2到第N+1行,每行有N个非负整数,第i+1 行第j个数表示Aij 。第N+2到第2*N+1行,每行有N个非负整数,第i+N+1 行第j个数表示Bij 。
输出格式:包含T!行,每行一个整数,表示最小的总体不和谐度
输入输出样例
说明
第1幅画搭配第3个画框,第2幅画搭配第1个画框,第3 幅画搭配第2个画框,则总体不和谐度为30
N<=70,T<=3,Aij<=200,Bij<=200
学了最小乘积生成树后发现这题很相似
只不过把生成树换成了KM还有这题求的是最小值,而KM求的是最大匹配
所以权要取负,算出来的点也取负
1 #include<iostream> 2 #include<cstdio> 3 #include<cmath> 4 #include<cstring> 5 #include<algorithm> 6 using namespace std; 7 typedef long long lol; 8 struct Node 9 { 10 lol x,y; 11 }; 12 Node operator -(Node u,Node v) 13 { 14 return (Node){u.x-v.x,u.y-v.y}; 15 } 16 Node operator +(Node u,Node v) 17 { 18 return (Node){u.x+v.x,u.y+v.y}; 19 } 20 lol operator *(Node u,Node v) 21 { 22 return u.x*v.y-u.y*v.x; 23 } 24 lol a[101][101],b[101][101],c[101][101]; 25 bool vis1[101],vis2[101]; 26 lol E1[101],E2[101],n,match[101],slack[101]; 27 lol ans; 28 bool dfs(int x) 29 {int i,j; 30 vis1[x]=1; 31 for (i=1;i<=n;i++) 32 if (vis2[i]==0) 33 { 34 lol gap=E1[x]+E2[i]-c[x][i]; 35 if (gap==0) 36 { 37 vis2[i]=1; 38 if (match[i]==-1||dfs(match[i])) 39 { 40 match[i]=x; 41 return 1; 42 } 43 } 44 else slack[i]=min(slack[i],gap); 45 } 46 return 0; 47 } 48 Node KM() 49 {int i,j; 50 lol tmp; 51 memset(match,-1,sizeof(match)); 52 for (i=1;i<=n;i++) 53 { 54 E2[i]=0; 55 tmp=0; 56 for (j=1;j<=n;j++) 57 { 58 if (c[i][j]>tmp) tmp=c[i][j]; 59 } 60 E1[i]=tmp; 61 } 62 for (i=1;i<=n;i++) 63 { 64 for (j=1;j<=n;j++) 65 slack[j]=1e15; 66 while (1) 67 { 68 memset(vis1,0,sizeof(vis1)); 69 memset(vis2,0,sizeof(vis2)); 70 if (dfs(i)) break; 71 lol tmp=1e15; 72 for (j=1;j<=n;j++) 73 if (vis2[j]==0&&tmp>slack[j]) tmp=slack[j]; 74 for (j=1;j<=n;j++) 75 { 76 if (vis1[j]) E1[j]-=tmp; 77 if (vis2[j]) E2[j]+=tmp; 78 else slack[j]-=tmp; 79 } 80 } 81 } 82 lol ansa=0,ansb=0; 83 for (i=1;i<=n;i++) 84 ansa+=a[match[i]][i],ansb+=b[match[i]][i]; 85 return (Node){-ansa,-ansb}; 86 } 87 lol cross(Node p,Node A,Node B) 88 { 89 return (A-p)*(B-p); 90 } 91 void solve(Node B,Node A) 92 {int i,j; 93 Node C; 94 lol x=(A.y-B.y),y=(B.x-A.x); 95 for (i=1;i<=n;i++) 96 for (j=1;j<=n;j++) 97 c[i][j]=-a[i][j]*x-b[i][j]*y; 98 C=KM(); 99 if (C.x*C.y<ans) ans=C.x*C.y; 100 if (cross(C,B,A)>=0) return; 101 solve(B,C); 102 solve(C,A); 103 } 104 int main() 105 {Node A,B; 106 int i,j,T; 107 cin>>T; 108 while (T--) 109 { 110 cin>>n; 111 ans=2e9; 112 for (i=1;i<=n;i++) 113 for (j=1;j<=n;j++) 114 scanf("%lld",&a[i][j]); 115 for (i=1;i<=n;i++) 116 for (j=1;j<=n;j++) 117 scanf("%lld",&b[i][j]); 118 for (i=1;i<=n;i++) 119 for (j=1;j<=n;j++) 120 c[i][j]=-b[i][j]; 121 A=KM(); 122 if (A.x*A.y<ans) ans=A.x*A.y; 123 for (i=1;i<=n;i++) 124 for (j=1;j<=n;j++) 125 c[i][j]=-a[i][j]; 126 B=KM(); 127 if (B.x*B.y<ans) ans=B.x*B.y; 128 solve(B,A); 129 printf("%lld\n",ans); 130 } 131 }