P r o b l e m \mathrm{Problem} Problem
小铭铭最近进入了某情报部门,该部门正在被如何建立安全的通道连接困扰。该部门有 n n n 个情报站,用 1 1 1 到 n n n 的整数编号。给出 m m m 对情报站 u i u_i ui; v i v_i vi 和费用 w i w_i wi,表示情报站 u i u_i ui 和 v i v_i vi 之间可以花费 w i w_i wi 单位资源建立通道。
如果一个情报站经过若干个建立好的通道可以到达另外一个情报站,那么这两个情报站就建立了通道连接。形式化地,若 u i u_i ui 和 v i v_i vi 建立了通道,那么它们建立了通道连接;若 u i u_i ui 和 v i v_i vi 均与 t i t_i ti 建立了通道连接,那么 u i u_i ui 和 v i v_i vi 也建立了通道连接。
现在在所有的情报站中,有 p p p 个重要情报站,其中每个情报站有一个特定的频道。小铭铭面临的问题是,需要花费最少的资源,使得任意相同频道的情报站之间都建立通道连接。
S o l u t i o n \mathrm{Solution} Solution
我们设 g i g_i gi表示连完点集为 i i i的最小花费。
我们一定可以由两个互不关联的连通块构成或者所有的点集都是一个连通块。
- 对于后者,我们可以直接由斯坦纳树解决。
- 对于前者,我们可以使用最基础的子集DP来搞定。
对于子集DP的转移方程就很显然了,斯坦纳树也是板子。
C o d e \mathrm{Code} Code
#include <bits/stdc++.h>
using namespace std;
const int N = 1100;
int n, m, p, cnt(0);
int f[N][1030], g[1030], Num[N], vis[N], c[N], d[N];
vector < int > TV[N];
vector < pair<int,int> > a[N];
priority_queue < pair<int,int> > q;
int read(void)
{
int s = 0, w = 0; char c = getchar();
while (c < '0' || c > '9') w |= c == '-', c = getchar();
while (c >= '0' && c <= '9') s = s * 10 + c - 48, c = getchar();
return w ? -s : s;
}
void Dijkstra(int S)
{
memset(vis,0,sizeof vis);
while (q.size())
{
int x = q.top().second; q.pop();
for (int i=0;i<a[x].size();++i)
{
int y = a[x][i].first;
int v = a[x][i].second;
if (f[x][S] + v < f[y][S]) {
f[y][S] = f[x][S] + v;
q.push({-f[y][S],y});
}
}
}
while (q.size()) q.pop();
return;
}
bool check(int S)
{
for (int i=1;i<=10;++i)
{
int sum = 0;
for (int j=0;j<TV[i].size();++j)
{
if (((S >> TV[i][j]) & 1) == 1) sum ++;
}
if (sum > 0 and sum < TV[i].size()) return 0;
}
return 1;
}
void DP(void)
{
int INF = f[0][0];
int MAX = (1 << cnt) - 1;
for (int i=1;i<=MAX;++i)
{
for (int j=1;j<=n;++j)
{
for (int k=i&(i-1);k;k=i&(k-1))
f[j][i] = min(f[j][i], f[j][k] + f[j][i-k]);
if (f[j][i] < INF) q.push({-f[j][i],j});
}
Dijkstra(i);
for (int j=1;j<=n;++j){
g[i] = min(g[i], f[j][i]);
}
}
g[0] = 0;
for (int i=1;i<=MAX;++i)
{
for (int j=(i-1)&i;j;j=(j-1)&i)
{
if (check(j) && check(i^j))
g[i] = min(g[i], g[j] + g[i^j]);
}
}
cout <<g[MAX] << endl;
return;
}
int main(void)
{
n = read(), m = read(), p = read();
for (int i=1;i<=m;++i)
{
int x = read(), y = read(), v = read();
a[x].push_back({y,v});
a[y].push_back({x,v});
}
memset(f,30,sizeof f);
memset(g,30,sizeof g);
for (int i=1;i<=p;++i)
{
c[i] = read(), d[i] = read();
if (Num[d[i]] == 0) Num[d[i]] = ++ cnt;
f[d[i]][1<<Num[d[i]]-1] = 0;
TV[c[i]].push_back(Num[d[i]]-1);
}
DP();
return 0;
}