【BZOJ4006】管道連接(動態規劃,斯坦納樹)
阿新 • • 發佈:2018-09-18
map 動態 code ring class new get efi include
題面
BZOJ
洛谷
題解
和這題區別不是很大吧。
基本上拿過來改一下就做完了。
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<algorithm> #include<vector> #include<queue> #include<map> using namespace std; #define ll long long #define MAX 1100 inline int read() { int x=0;bool t=false;char ch=getchar(); while((ch<'0'||ch>'9')&&ch!='-')ch=getchar(); if(ch=='-')t=true,ch=getchar(); while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar(); return t?-x:x; } int n,m,p; struct Line{int v,next,w;}e[MAX<<3]; int h[MAX],cnt=1; inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;} int f[1<<10][MAX],g[1<<10]; bool vis[MAX]; queue<int> Q; map<int,int> M; int fz[MAX],St[20],G[20]; void SPFA(int *f) { while(!Q.empty()) { int u=Q.front();Q.pop(); for(int i=h[u];i;i=e[i].next) if(f[e[i].v]>f[u]+e[i].w) { f[e[i].v]=f[u]+e[i].w; if(!vis[e[i].v])Q.push(e[i].v),vis[e[i].v]=true; } vis[u]=false; } } bool check(int s) { for(int i=1;i<=p;++i) if((s&G[i])!=0&&(s&G[i])!=G[i]) return false; return true; } int main() { n=read();m=read();p=read(); memset(f,63,sizeof(f));memset(g,63,sizeof(g)); for(int i=1;i<=m;++i) { int u=read(),v=read(),w=read(); Add(u,v,w);Add(v,u,w); } for(int i=1;i<=p;++i) { int c=read(),d=read(); fz[d]=c;M[d]=i-1;St[i]=d; f[1<<(i-1)][d]=0; } for(int i=1;i<=p;++i) for(int j=1;j<=p;++j) if(fz[St[i]]==fz[St[j]]) G[i]|=1<<M[St[j]]; int S=1<<p; for(int i=0;i<S;++i) { for(int j=1;j<=n;++j) { for(int k=i&(i-1);k;k=(k-1)&i) f[i][j]=min(f[i][j],f[k][j]+f[i^k][j]); if(f[i][j]<=1e9)Q.push(j),vis[j]=true; } SPFA(f[i]); for(int j=1;j<=n;++j)g[i]=min(g[i],f[i][j]); } for(int i=0;i<S;++i) for(int t=(i-1)&i;t;t=(t-1)&i) if(check(t)&&check(i^t)) g[i]=min(g[i],g[t]+g[i^t]); printf("%d\n",g[S-1]<=1e9?g[S-1]:-1); return 0; }
【BZOJ4006】管道連接(動態規劃,斯坦納樹)