斯坦纳树:
给n个点m条边的无向图,然后给出k个点的集合,找一个最小的生成树使得包含这k个点(这个生成树必须包含这k个点,其他点可以有也可以没有,但是必须保证最后是个生成树,然后找最小的生成树)。
利用状压加最短路算法实现,目前没有多项式时间复杂度的算法。
dis[x][p]代表以x为根,k个点集合中选择了状态为p的点的最小生成树权值。
两个核心转移方程 dis[x][p]=min(dis[x][p],dis[x][i]+dis[x][p^i]);
和 dis[x][p]=min(dis[x][p],dis[y][p]+cost(x,y));
第一个方程通过枚举p以及p的子集加上枚举n个点实现,复杂度是
n
∗
3
k
n*3^k
n∗3k。
第二个方程通过最短路实现即可,最短路转移时,p不改变,因此和正常的最短路算法一样,只不过多了个p而已,复杂度是
2
k
∗
2^k*
2k∗o(最短路)。
总的复杂度即为
n
∗
3
k
+
2
k
∗
n*3^k+2^k*
n∗3k+2k∗o(最短路)。
初始化为把所有的dp[x][0或者1<<x]初始化为0,其余为INF。
例题:魔物消灭计划
这个题相当于模板题。
spfa写法:(如果有负权必须用spfa)
#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int MAX_N=110;
const int INF=0x3f3f3f3f;
int a[MAX_N],id[MAX_N];
vector<int>v[MAX_N],w[MAX_N];
queue<int>q;
bool vis[MAX_N];
int dis[MAX_N][1100];
int k;
void spfa(int p){
int i;
while(!q.empty()){
int x=q.front();
q.pop();
for(i=0;i<v[x].size();i++){
int y=v[x][i];
int cost=w[x][i];
if(dis[y][p]>dis[x][p]+cost){
dis[y][p]=dis[x][p]+cost;
if(!vis[y]){
vis[y]=true;
q.push(y);
}
}
}
vis[x]=false;
}
}
int main(void){
int n,m,x,y,i,j,xx,yy,zz;
scanf("%d%d%d%d%d",&n,&m,&k,&x,&y);
memset(dis,INF,sizeof(dis));
int st=k+1,ed=k+2;
id[x]=st,id[y]=ed;
int cnt=k+2;
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
if(i==x||i==y)
continue;
if(a[i]==0)
id[i]=++cnt;
else
id[i]=a[i];
}
for(i=1;i<=m;i++){
scanf("%d%d%d",&xx,&yy,&zz);
if(xx==yy)
continue;
xx=id[xx],yy=id[yy];
v[xx].push_back(yy);
w[xx].push_back(zz);
v[yy].push_back(xx);
w[yy].push_back(zz);
}
int tt=(1<<(k+2)),t;
for(i=1;i<=k+2;i++)
dis[i][1<<(i-1)]=0;
for(i=k+3;i<=cnt;i++)
dis[i][0]=0;
for(i=0;i<tt;i++){
for(j=i;j;j=(j-1)&i){
for(t=1;t<=cnt;t++){
dis[t][i]=min(dis[t][i],dis[t][j]+dis[t][i^j]);
}
}
for(j=1;j<=cnt;j++){
if(dis[j][i]!=INF)
q.push(j),vis[j]=true;
}
spfa(i);
}
printf("%d\n",dis[st][tt-1]);
return 0;
}
dijkstra写法:
#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int MAX_N=110;
const int INF=0x3f3f3f3f;
int a[MAX_N],id[MAX_N];
vector<int>v[MAX_N],w[MAX_N];
bool vis[MAX_N];
int dis[MAX_N][1100];
int k;
struct skt{
int x,d;
bool friend operator<(skt a,skt b){
return a.d>b.d;
}
};
priority_queue<skt>q;
void dijkstra(int p){
int i;
while(!q.empty()){
skt t=q.top();
q.pop();
int x=t.x;
if(vis[x])
continue;
vis[x]=true;
for(i=0;i<v[x].size();i++){
int y=v[x][i];
int cost=w[x][i];
if(dis[y][p]>dis[x][p]+cost){
dis[y][p]=dis[x][p]+cost;
t.x=y,t.d=dis[y][p];
q.push(t);
}
}
}
}
int main(void){
int n,m,x,y,i,j,xx,yy,zz;
scanf("%d%d%d%d%d",&n,&m,&k,&x,&y);
memset(dis,INF,sizeof(dis));
int st=k+1,ed=k+2;
id[x]=st,id[y]=ed;
int cnt=k+2;
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
if(i==x||i==y)
continue;
if(a[i]==0)
id[i]=++cnt;
else
id[i]=a[i];
}
for(i=1;i<=m;i++){
scanf("%d%d%d",&xx,&yy,&zz);
if(xx==yy)
continue;
xx=id[xx],yy=id[yy];
v[xx].push_back(yy);
w[xx].push_back(zz);
v[yy].push_back(xx);
w[yy].push_back(zz);
}
int tt=(1<<(k+2)),t;
for(i=1;i<=k+2;i++)
dis[i][1<<(i-1)]=0;
for(i=k+3;i<=cnt;i++)
dis[i][0]=0;
for(i=0;i<tt;i++){
for(j=i;j;j=(j-1)&i){
for(t=1;t<=cnt;t++){
dis[t][i]=min(dis[t][i],dis[t][j]+dis[t][i^j]);
}
}
for(j=1;j<=cnt;j++){
vis[j]=false;
if(dis[j][i]!=INF){
skt t;t.x=j,t.d=dis[j][i];
q.push(t);
}
}
dijkstra(i);
}
printf("%d\n",dis[st][tt-1]);
return 0;
}