挑战NPC
描述
小 N 最近在研究 NP 完全问题,小 O 看小 N 研究得热火朝天,便给他出了一道这样的题目:
有 nn 个球,用整数 11 到 nn 编号。还有 mm 个筐子,用整数 11 到 mm 编号。
每个筐子最多能装 3 个球。
每个球只能放进特定的筐子中。具体有 ee 个条件,第 ii 个条件用两个整数 vivi 和 uiui 描述,表示编号为 vivi 的球可以放进编号为 uiui 的筐子中。
每个球都必须放进一个筐子中。如果一个筐子内有不超过 11 个球,那么我们称这样的筐子为半空的。
求半空的筐子最多有多少个,以及在最优方案中,每个球分别放在哪个筐子中。
小 N 看到题目后瞬间没了思路,站在旁边看热闹的小 I 嘿嘿一笑:“水题!”
然后三言两语道出了一个多项式算法。
小 N 瞬间就惊呆了,三秒钟后他回过神来一拍桌子:
“不对!这个问题显然是 NP 完全问题,你算法肯定有错!”
小 I 浅笑:“所以,等我领图灵奖吧!”
小 O 只会出题不会做题,所以找到了你——请你对这个问题进行探究,并写一个程序解决此题。
输入格式
第一行包含 1 个正整数 T,表示有 T 组数据。
对于每组数据,第一行包含 3 个正整数 n,m,e,表示球的个数,筐子的个数和条件的个数。
接下来 e 行,每行包含 2 个整数 vi,ui,表示编号为 vi 的球可以放进编号为 ui 的筐子。
输出格式
对于每组数据,先输出一行,包含一个整数,表示半空的筐子最多有多少个。
然后再输出一行,包含 n 个整数 p1,p2,…,pn,相邻整数之间用空格隔开,表示一种最优解。其中 pi 表示编号为 ii 的球放进了编号为 pi 的筐子。如果有多种最优解,可以输出其中任何一种。
样例一
input
1
4 3 6
1 1
2 1
2 2
3 2
3 3
4 3
output
2
1 2 3 3
样例二
见样例数据下载。
限制与约定
对于所有数据,T≤5,1≤n≤3m。保证 1≤vi≤n,1≤ui≤m,且不会出现重复的条件。
保证至少有一种合法方案,使得每个球都放进了筐子,且每个筐子内球的个数不超过 3。
时间限制:1s
空间限制:256MB
一道建图十分巧妙的题目~
另外这里没有带花树,想学带花树的请出门左转。
思路:
一个很巧妙的建图啊。
建图方式:
把每个筐子拆成3个点,从中任选两个点并在这两个点之间连边。
每个球建一个点,向能放的筐子的三个点分别连一条边。
按先增广球,后增广筐子的顺序跑一般图最大匹配,最后最大匹配数-n便是答案~
为什么这是正确的呢?
考虑只有一个筐子时的情况。
当没有球时,最大匹配数为1,ans=1-0=1。
当有一个球时,最大匹配数为2,ans=2-1=1。
当有两个球时,最大匹配数为2,ans=2-2=0。
当有三个球时,最大匹配数为3,ans=3-3=0。
神奇的对上了!
事实上,筐内部的连边代表半空,球向筐的连边代表放入一个球。
然后就是跑一般图最大匹配了~
然而本蒟蒻不会带花树,于是写了个随机化匈牙利~
这个随机匈牙利的方法如下:
用邻接表存图。
每次dfs到一个点,先random_shuffle一下儿子,再去增广。
这样直接跑即可!
实测正确率十分之高~
然而某不知名dalao使用这个算法在一般图最大匹配那题被人Hack了。
被Hack成功的原因是这位同志运气很不好
于是改进一下重复跑5次取最优解,然后就可以稳过Hack了!
妈妈再也不用担心咱不会写带花树了~
目前这题UOJ的最短榜rk1是咱的小号
#include <bits/stdc++.h>
using namespace std;
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0' || '9'<ch)ch=getchar();
while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
return x;
}
namespace fake_blossom
{
const int N=609;
vector<int> to[N];
int n,match[N],vis[N],tim=0;
int hmatch[N],cans,ans;
inline void init(int p)
{
n=p;
for(int i=0;i<N;i++)
to[i].clear();
memset(hmatch,0,sizeof(hmatch));
memset(vis,0,sizeof(vis));
cans=tim=0;
}
inline void add(int u,int v)
{
to[u].push_back(v);
to[v].push_back(u);
}
bool dfs(int x)
{
if(!x)return 1;
vis[x]=tim;
random_shuffle(to[x].begin(),to[x].end());
for(int i=0,u,v;i<to[x].size();i++)
{
u=to[x][i];
if(vis[v=match[u]]<tim)
{
match[x]=u,match[u]=x,match[v]=0;
if(dfs(v))return true;
match[u]=v,match[v]=u,match[x]=0;
}
}
return false;
}
inline void work()
{
int T=5;cans=-1;
while(T--)
{
memset(match,0,sizeof(match));
ans=0;
for(int i=1;i<=n;i++)
if(!match[i])
tim++,ans+=dfs(i);
if(ans>cans)
cans=ans,memcpy(hmatch,match,sizeof(match));
}
}
}
inline void mina()
{
int n,m,e,ans=0;
n=read();m=read();e=read();
srand(time(0));
fake_blossom::init(n+m*3);
for(int i=1,u,v;i<=e;i++)
{
u=read();v=read();
for(int j=0;j<=2;j++)
fake_blossom::add(u,n+v*3-j);
}
for(int i=1;i<=m;i++)
fake_blossom::add(n+i*3-1,n+i*3);
fake_blossom::work();
for(int i=1;i<=n+m*3;i++)
ans+=(fake_blossom::hmatch[i]>0);
printf("%d\n",ans/2-n);
for(int i=1;i<=n;i++)
printf("%d ",(fake_blossom::hmatch[i]-n+2)/3);
puts("");
}
int main()
{
int T=read();
while(T--)
mina();
return 0;
}