题目大意
给你一个n(n<=100)点的有向带权图,你需要找若干个圈,使得每个节点恰好属于一个圈(不可相交),如果有这样的方案,请输出方案的最小权值和。注意是有向图。
思路
每一个点属于一个圈,这可以想到DAG的最小路径覆盖,每个点属于一条路径,在DAG的最小路径覆盖中,找到的最大匹配数实际是非终点的点的最大数量(每一个匹配对应一个起点),点数减去这个数量就是终点的最少数量,一个终点对应一条路径,所以这个路径覆盖是最小的。
那么此题中,对于每个点,如果在一个圈内,,也就意味这这个圈上的每个点都可以是这个圈的起点,即每个点都是匹配的,也就是终点的答案是0,所以只要图存在完美匹配,那么解是存在的。
首先,用匈牙利算法寻找一下最大匹配,如果等于n,则有完美匹配,再求权值最小的完美匹配,就可以像DAG最小路径覆盖那样,对于每个点,拆成两个点U_x,U_y,如果有边u_v,建边U_x->V_y,用KM算法求最小权值匹配即可。
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 105;
const int inf = 0x3f3f3f3f;
int match[N], mp[N][N];
int link[N];
bool vis[N];
bool vis_x[N], vis_y[N];
int slack[N];
int lx[N], ly[N];
vector<int> g[N];
bool dfs(int u)
{
vis_x[u] = 1;
for (int j = 0; j < g[u].size(); j++)
{
int i = g[u][j];
if (vis_y[i]) continue;
int gap = lx[u] + ly[i] - mp[u][i];
if (gap == 0)
{
vis_y[i] = 1;
if (match[i] == -1 || dfs(match[i]))
{
match[i] = u;
return true;
}
}
else
{
slack[i] = min(gap, slack[i]);
}
}
return false;
}
int km(int n)
{
memset(match, -1, sizeof(match));
memset(ly, 0, sizeof(ly));
for (int i = 1; i <= n; i++)
{
lx[i] = mp[i][1];
for (int j = 2; j <= n; j++)
{
lx[i] = max(lx[i], mp[i][j]);
}
}
for (int i = 1; i <= n; i++)
{
memset(slack, inf, sizeof(slack));
while (1)
{
memset(vis_x, 0, sizeof(vis_x));
memset(vis_y, 0, sizeof(vis_y));
if (dfs(i)) break;
int d = inf;
for (int j = 1; j <= n; j++)
{
if (!vis_y[j]) d = min(d, slack[j]);
}
for (int j = 1; j <= n; j++)
{
if (vis_x[j]) lx[j] -= d;
if (vis_y[j]) ly[j] += d;
}
}
}
int ans = 0;
for (int i = 1; i <= n; i++)
{
ans += mp[match[i]][i];
}
return ans;
}
bool dfs2(int u)
{
for (int i = 0; i < g[u].size(); i++)
{
int v = g[u][i];
if (vis[v]) continue;
vis[v] = 1;
if (link[v] == -1 || dfs2(link[v]))
{
link[v] = u;
return true;
}
}
return false;
}
bool hungary(int n)
{
memset(link, -1, sizeof(link));
int res = 0;
for (int i = 1; i <= n; i++)
{
memset(vis, 0, sizeof(vis));
if (dfs2(i)) res++;
}
return res >= n;
}
int main()
{
int n;
while (scanf("%d", &n) != EOF && n)
{
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
mp[i][j] = -inf;
}
for (int i = 0; i <= n; i++) g[i].clear();
for (int i = 1; i <= n; i++)
{
int a, b;
while (scanf("%d", &a) != EOF)
{
if (a == 0) break;
scanf("%d", &b);
mp[i][a] = -b;
g[i].push_back(a);
}
}
if (hungary(n))
{
printf("%d\n", -km(n));
}
else
{
printf("N\n");
}
}
return 0;
}