【BZOJ4774】修路

Description

村子间的小路年久失修,为了保障村子之间的往来,法珞决定带领大家修路。对于边带权的无向图 G = (V, E),请选择一些边,使得1 <= i <= d, i号节点和 n - i + 1 号节点可以通过选中的边连通,最小化选中的所有边的权值和。

Input

第一行两个整数 n, m,表示图的点数和边数。接下来的 m行,每行三个整数 ui, vi, wi,表示有一条 ui 与 vi 之间,权值为 wi 的无向边。
1 <= d <= 4
2d <= n <= 10^4
0 <= m <= 10^4
1 <= ui, vi <= n
1 <= wi <= 1000

Output

一行一个整数,表示答案,如果无解输出-1

Sample Input

10 20 1
6 5 1
6 9 4
9 4 2
9 4 10
6 1 2
2 3 6
7 6 10
5 7 1
9 7 2
5 9 10
1 6 8
4 7 4
5 7 1
2 6 9
10 10 6
8 7 2
10 9 10
1 2 4
10 1 8
9 9 7

Sample Output

8

题解:设f[S][i]表示已经连通的关键点状态为S,当前位于点i的最小权值和。转移就是斯坦纳树。

再设g[S]表示已经连通的关键点状态为S的最小权值和。只有当S保证所有的关键点对的连通状态相同时,才可以从f[S][..]更新到g[S],最后对g数组跑枚举子集的DP即可。

P.S.我至今才会枚举子集的正确姿势~~~

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <utility>
#define mp(A,B) make_pair(A,B)
using namespace std;
int n,m,d,cnt,now,tot;
int f[1<<8][10010],to[20010],next[20010],val[20010],head[10010],vis[10010],p[20],Log[1<<8],ref[1<<8],g[1<<8];
priority_queue<pair<int,int> > q;
inline int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<'0'||gc>'9') {if(gc=='-') f=-f; gc=getchar();}
while(gc>='0'&&gc<='9') ret=ret*10+(gc^'0'),gc=getchar();
return ret*f;
}
inline void add(int a,int b,int c)
{
to[cnt]=b,val[cnt]=c,next[cnt]=head[a],head[a]=cnt++;
}
inline void dij(int S,int x)
{
q.push(mp(-f[S][x],x));
int i,u;
now++;
while(!q.empty())
{
u=q.top().second,q.pop();
if(vis[u]==now) continue;
vis[u]=now;
for(i=head[u];i!=-1;i=next[i]) if(f[S][to[i]]>f[S][u]+val[i])
f[S][to[i]]=f[S][u]+val[i],q.push(mp(-f[S][to[i]],to[i]));
}
}
int main()
{
int i,a,b,c,x,y;
n=rd(),m=rd(),d=rd();
memset(head,-1,sizeof(head));
for(i=1;i<=m;i++) a=rd(),b=rd(),c=rd(),add(a,b,c),add(b,a,c);
for(i=0;i<2*d;i++) Log[1<<i]=i;
memset(f,0x3f,sizeof(f)),memset(g,0x3f,sizeof(g));
for(i=1;i<=d;i++)
{
f[1<<(i-1)][i]=0,dij(1<<(i-1),i);
f[1<<(d+i-1)][n-i+1]=0,dij(1<<(d+i-1),n-i+1);
}
for(x=1;x<(1<<(2*d));x++)
{
for(tot=0,y=x;y;y-=y&-y) p[tot++]=y&-y;
for(y=1;y<(1<<tot);y++)
{
ref[y]=ref[y^(y&-y)]|p[Log[y&-y]];
for(i=1;i<=n;i++) if(f[x][i]>f[ref[y]][i]+f[x^ref[y]][i])
f[x][i]=f[ref[y]][i]+f[x^ref[y]][i];
}
for(i=1;i<=n;i++) dij(x,i);
}
for(x=1;x<(1<<d);x++) for(i=1;i<=n;i++) g[x]=min(g[x],f[(x<<d)|x][i]);
for(x=1;x<(1<<d);x++)
{
for(tot=0,y=x;y;y-=y&-y) p[tot++]=y&-y;
for(y=1;y<(1<<tot);y++)
{
ref[y]=ref[y^(y&-y)]|p[Log[y&-y]];
g[x]=min(g[x],g[ref[y]]+g[x^ref[y]]);
}
}
if(g[(1<<d)-1]==0x3f3f3f3f) printf("-1");
else printf("%d",g[(1<<d)-1]);
return 0;
}
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <utility>
#define mp(A,B) make_pair(A,B)
using namespace std;
int n,m,d,cnt,now;
int to[6010],next[6010],val[6010],head[1010],f[1<<10][1010],g[1<<10],p[15],w[15],vis[1<<10];
priority_queue<pair<int,int> > q;
inline int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<'0'||gc>'9') {if(gc=='-') f=-f; gc=getchar();}
while(gc>='0'&&gc<='9') ret=ret*10+(gc^'0'),gc=getchar();
return ret*f;
}
inline void add(int a,int b,int c)
{
to[cnt]=b,val[cnt]=c,next[cnt]=head[a],head[a]=cnt++;
}
inline void dij(int S,int x)
{
q.push(mp(-f[S][x],x));
int i,u;
now++;
while(!q.empty())
{
u=q.top().second,q.pop();
if(vis[u]==now) continue;
vis[u]=now;
for(i=head[u];i!=-1;i=next[i]) if(f[S][to[i]]>f[S][u]+val[i])
f[S][to[i]]=f[S][u]+val[i],q.push(mp(-f[S][to[i]],to[i]));
}
}
int main()
{
n=rd(),m=rd(),d=rd();
int i,j,a,b,c,x,y;
memset(head,-1,sizeof(head));
for(i=1;i<=m;i++) a=rd(),b=rd(),c=rd(),add(a,b,c),add(b,a,c);
memset(f,0x3f,sizeof(f)),memset(g,0x3f,sizeof(g));
for(i=0;i<d;i++) w[i]=rd(),p[i]=rd(),f[1<<i][p[i]]=0,dij(1<<i,p[i]);
for(x=1;x<(1<<d);x++)
{
for(y=(x-1)&x;y;y=(y-1)&x) for(i=1;i<=n;i++) f[x][i]=min(f[x][i],f[y][i]+f[x^y][i]);
for(i=1;i<=n;i++) dij(x,i);
}
for(x=1;x<(1<<d);x++) for(i=1;i<=n;i++) g[x]=min(g[x],f[x][i]);
for(x=1;x<(1<<d);x++)
{
for(i=0;i<d;i++) for(j=0;j<i;j++) if(w[i]==w[j]&&((x>>i)&1)!=((x>>j)&1)) g[x]=0x3f3f3f3f;
}
for(x=1;x<(1<<d);x++) for(y=(x-1)&x;y;y=(y-1)&x) g[x]=min(g[x],g[y]+g[x^y]);
printf("%d",g[(1<<d)-1]);
return 0;
}
05-11 17:46