2019.03.26【JLOI2015】【洛谷P3264】【BZOJ4006】【LOJ2110】管道连接(斯坦纳树)(状压DP)

版权声明:转载请声明出处,谢谢配合。 https://blog.csdn.net/zxyoi_dreamer/article/details/88816477

洛谷传送门

BZOJ传送门

LOJ传送门


解析:

很显然,由于求的是令某些特殊点连接起来的最小代价,所以解应该是斯坦纳森林。

先求出所有状态的斯坦纳树,然后枚举集合的划分,check一下同一频道的点是否全在或全不在当前状态中,之后直接转移就好了。


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<20|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++; 
	}
	
	inline int getint(){
		re char c;
		while(!isdigit(c=gc()));re int num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
}
using namespace IO;

using std::cout;
using std::cerr;
using std::swap;

typedef std::pair<int,int> pii;
#define mp std::make_pair

cs int N=1003; 
int n,m,p;

struct edge{
	int to,w;
	edge(cs int &_to,cs int &_w):to(_to),w(_w){}
};

std::vector<edge> e[N];
inline void addedge(int u,int v,int w){
	e[u].push_back(edge(v,w));
	e[v].push_back(edge(u,w));
} 

cs int INF=1e9;
int f[1<<10][N],g[1<<10];
std::priority_queue<pii,std::vector<pii>,std::greater<pii> > q;
bool vis[N];
inline void Dij(int S){
	int *dist=f[S];
	memset(vis+1,0,sizeof(bool)*n);
	while(!q.empty()){
		int u=q.top().second;q.pop();
		if(vis[u])continue;
		vis[u]=true;
		for(std::vector<edge>::iterator it=e[u].begin();it!=e[u].end();++it){
			if(dist[it->to]>dist[u]+it->w)q.push(mp(dist[it->to]=dist[u]+it->w,it->to));
		} 
	}
} 

int id[11],c[11],tot[11];
bool ok[1<<10];

inline bool check(int S){
	static int cnt[11];
	memset(cnt,0,sizeof cnt);
	for(int re i=1;i<=p;++i)if(S&(1<<(i-1)))++cnt[c[i]];
	for(int re i=1;i<=10;++i)if(cnt[i]&&cnt[i]!=tot[i])return false;
	return true;
} 

signed main(){
	memset(f,0x3f,sizeof f);
	memset(g,0x3f,sizeof g);
	n=getint(),m=getint(),p=getint();
	for(int re i=1;i<=m;++i){
		int u=getint(),v=getint(),w=getint();
		addedge(u,v,w);
	}
	for(int re i=1;i<=p;++i){
		c[i]=getint(),id[i]=getint();
		++tot[c[i]];
		f[1<<(i-1)][id[i]]=0;
	}
	for(int re s=1;s<(1<<p);++s){
		for(int re i=1;i<=n;++i){
			for(int re j=s&(s-1);j;j=s&(j-1))
			f[s][i]=std::min(f[s][i],(f[s^j][i]+f[j][i]));
			if(f[s][i]<INF)q.push(mp(f[s][i],i));
		}
		Dij(s);
		for(int re i=1;i<=n;++i)g[s]=std::min(g[s],f[s][i]);
		ok[s]=check(s);
	}
	for(int re s=1;s<(1<<p);++s)
	if(ok[s])for(int re j=s&(s-1);j;j=s&(j-1))if(ok[j])g[s]=std::min(g[s],g[s^j]+g[j]);
	cout<<g[(1<<p)-1]<<"\n";
	return 0;
}

猜你喜欢

转载自blog.csdn.net/zxyoi_dreamer/article/details/88816477
今日推荐