题目链接:https://vjudge.net/problem/Gym-101873F
题意:输入m,n,k。代表m个插座,n个设备,k行输入。接下来每行输入a,b代表a插座可以被b设备使用。并且有一个转换器可以让一个插座最多被3个用电器使用。求最多可以连接的设备数目。
思路:匈牙利算法求最大匹配这个是肯定的,问题在于如何枚举。如果先枚举每个插座用转换器时的情况,在遍历跑匈牙利是超时的。这时候可以先假设没有转换器时跑一边匈牙利,保存匹配结果,可以节省大量时间,然后对每一个插座看是否还存在增广路。
参考:https://blog.csdn.net/monochrome00/article/details/82946070
#include <bits/stdc++.h>
using namespace std;
vector <int> g[1505];
int vis[1505];
int match[1505],match2[1505];
int n;
bool dfs(int x)
{
int len=g[x].size();
for(int i=0; i<len; i++)
{
int t=g[x][i];
if(!vis[t])
{
vis[t]=1;
if(!match[t]||dfs(match[t]))
{
match[t]=x;
return true;
}
}
}
return false;
}
int main()
{
int m,n,k,ans=0;
scanf("%d%d%d",&m,&n,&k);
for(int i=1; i<=1500; i++)
{
g[i].clear();
}
int a,b;
for(int i=0; i<k; i++)
{
scanf("%d%d",&a,&b);
g[a].push_back(b);
}
for(int i=1; i<=m; i++)
{
memset(vis,0,sizeof(vis));
if(dfs(i))
{
ans++;
}
}
memcpy(match2,match,sizeof(match));
int cnt=0;
for(int i=1; i<=m; i++)
{
int cnt1=0;
for(int j=1; j<=2; j++)
{
memset(vis,0,sizeof(vis));
if(dfs(i))
cnt1++;
}
cnt=max(cnt,cnt1);
memcpy(match,match2,sizeof(match2));
}
printf("%d\n",ans+cnt);
return 0;
}