[ABC318D] General Weighted Max Matching 题解
[ABC318D] General Weighted Max Matching 题解
题意
给定无向有权完全图,求最大权匹配。
思路分析
注意到 \(n \le 16\),我考虑状压 DP。
设当前点集 \(S\) 中最大权匹配的答案是 \(f_S\),我们考虑 \(S\) 中“最后”一个点 \(p\)(这里的“最后”一个点是指,在状压表示状态的时候,最后一个 1 所代表的那个点,只需从这个点考虑就行,不需要考虑其他前面的点,因为会被更小状态考虑过)。
我们可以从前面其他点中,选择一个点 \(q\) 和这个点匹配,也可以不匹配这个点。于是有转移方程:
\[f_S = \max(f_{S-p},f_{S-p-q}),p \in S,q \in S, p \ne q
\]
其中 \(w_{p,q}\) 代表点 \(p\) 与点 \(q\) 之间的边权。
初始化:当 \(S\) 中没有点或只有一个点的时候 \(f_S = 0\);当 \(S\) 中有两个点 \(u,v\) 时,\(F_S = w_{u,v}\)。
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 18;
const int S = (1<<(N+1));
LL a[N][N];
LL dp[S];
LL cnt[S];
LL n;
int count(int s)
{
if(cnt[s] != -1)
return cnt[s];
int ans = 0;
while(s>0)
{
ans += (s&1);
s >>= 1;
}
return cnt[s] = ans;
}
LL dfs(int s)
{
if(dp[s] != -1) return dp[s];
if(count(s) <= 1)
{
dp[s] = 0;
return 0;
}
if(count(s) == 2)
{
int idx1 = -1,idx2 = -1;
int tmp = s,tmpidx = 0;
while(tmp>0)
{
if(tmp&1)
{
if(idx1 == -1)
{
idx1 = tmpidx;
}
else
{
idx2 = idx1;
idx1 = tmpidx;
}
}
tmpidx++;
tmp >>= 1;
}
dp[s] = a[idx1][idx2];
return dp[s];
}
LL idxf = -1;
LL ans = 0;
for(int i = n-1;i >= 0;i--)
{
if((s&(1<<i)) > 0)
{
if(idxf == -1)
{
idxf = i;
}
else
{
ans = max(ans,a[idxf][i]+dfs(s-(1<<i)-(1<<idxf)));
}
}
}
ans = max(ans,dfs(s-(1<<idxf)));
return dp[s] = ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
memset(dp,-1,sizeof(dp));
memset(cnt,-1,sizeof(cnt));
cin >> n;
for(int i = 1;i <= n;i++)
{
for(int j = i+1;j <= n;j++)
{
cin >> a[i-1][j-1];
a[j-1][i-1] = a[i-1][j-1];
}
}
dfs((1<<n)-1);
cout << dp[(1<<n)-1] << "\n";
return 0;
}