SGU 210 Beloved Sons （KM）

来源：互联网发布：淘宝企业开店编辑：程序博客网时间：2024/05/01 23:50

传送门：SGU 210 Beloved Sons

题意：二分图x部分是王子，y部分是女孩，王子要从中挑选一个他喜欢的结婚，每个王子受国王的宠爱程度不同，宠爱度越高的王子结婚国王越高兴，要求让国王最高兴。

g[x][y]代表二分图的边，match1[x]代表与x匹配的y，match2[y]代表与y匹配的x。

/*************************************************************************     File Name: B.cpp     ID: obsoles1     PROG:      LANG: C++      Mail: 384099319@qq.com      Created Time: 一  9/ 5 23:26:53 2016 ************************************************************************/#define MAX(x,y) ((x)>(y)?(x):(y))#define MIN(x,y) ((x)<(y)?(x):(y))#define EACH(it,v) for(__typeof((v).begin()) it=(v).begin();it!=(v).end();++it)#define ABS(x,y) ((x)>(y)?((x)-(y)):((y)-(x)))#define pb push_back#define MEM(a,x) memset(a,x,sizeof(a))#include<cstdio>#include<cstring>using namespace std;typedef long long LL;const int N=410;const int INF=0x3f3f3f3f;const int MOD=1000000007;int g[N][N],l1[N],l2[N],s[N],t[N],match1[N],match2[N],n,ans[N];void KM() {  int p,q,i,j,k;  for (i=0; i < n; ++i) {    l1[i]=-INF;    for (j=0; j < n; ++j)      l1[i]=g[i][j]>l1[i]? g[i][j]:l1[i];  }  MEM(l2,0),MEM(match1,-1),MEM(match2,-1);  for (i=0; i < n; ++i) {    MEM(t,-1),p=q=0;    for (s[0]=i; p <= q && match1[i] < 0; ++p)      for (k=s[p],j=0; j < n && match1[i] < 0; ++j) {        if (l1[k]+l2[j]!=g[k][j] || t[j] >= 0) continue;        s[++q]=match2[j];        t[j]=k;        if (s[q] >= 0) continue;        for (p=j; p >= 0; j=p) {          match2[j]=k=t[j];          p=match1[k];          match1[k]=j;        }      }    if (match1[i] >= 0) continue;    --i,p=INF;    for (k=0; k <= q; ++k)      for (j=0; j < n; ++j)        if (t[j] < 0 && l1[s[k]]+l2[j]-g[s[k]][j] < p)          p=l1[s[k]]+l2[j]-g[s[k]][j];    for (j=0; j < n; ++j)      l2[j]+=t[j]<0? 0:p;    for (k=0; k <= q; ++k)      l1[s[k]]-=p;  }  MEM(ans,-1);    for (i=0; i < n; ++i) {    if (g[i][match1[i]])      printf("%d%c",match1[i]+1," \n"[i==n-1]);    else printf("0%c"," \n"[i==n-1]);  }}int v[N];int main() {  int k,i,j,x;  while (~scanf("%d",&n)) {    MEM(g,0);    for (i=0; i < n; ++i)      scanf("%d",v+i);    for (i=0; i < n; ++i) {      scanf("%d",&k);      for (j=0; j < k; ++j) {        scanf("%d",&x);        x--;        g[i][x]=v[i];      }    }    KM();  }}

0 0