CF1005F Berland and the Shortest Paths 最短路树计数

问题描述

LG-CF1005F

---

题解

由题面显然可得，所求即最短路树。

所以跑出最短路树，计数，输出方案即可。

---

\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

template <typename Tp>
void read(Tp &x){
	x=0;char ch=1;int fh;
	while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
	if(ch=='-') ch=getchar(),fh=-1;
	else fh=1;
	while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
	x*=fh;
}

const int maxn=200007;
const int maxm=400007;
const int INF=0x3f3f3f3f;

int n,m,k;
int Head[maxn],to[maxm],Next[maxm],tot,w[maxm];

void add(int x,int y){
	to[++tot]=y,Next[tot]=Head[x],Head[x]=tot,w[tot]=1;
}

int dis[maxn];
bool vis[maxn];

priority_queue< pair<int,int> > q;
#define pii(x,y) make_pair(x,y)

void dijkstra(){
	memset(dis,0x3f,sizeof(dis));
	q.push(pii(0,1));dis[1]=0;
	while(q.size()){
		int x=q.top().second;q.pop();
		if(vis[x]) continue;vis[x]=1;
		for(int i=Head[x];i;i=Next[i]){
			int y=to[i];
			if(dis[y]>dis[x]+w[i]){
				dis[y]=dis[x]+w[i];
				q.push(pii(-dis[y],y));
			}
		}
	}
}

int ans=1;

vector<int>g[maxn];
int val[maxn];
void build(){
	for(int x=1;x<=n;x++){
		for(int i=Head[x];i;i=Next[i]){
			int y=to[i];
			if(dis[y]==dis[x]+w[i]){
				val[y]++;
				g[y].push_back((i+1)>>1);
			}
		}
	}
	for(int i=1;i<=n;i++){
		if(val[i]) ans=ans*val[i];
		if(ans>=k){
			ans=k;return;
		}
	}
}

bool v[maxm];
int md;

void dfs(int x){
	if(x==n+1){
		for(int i=1;i<=tot;i+=2) printf("%d",v[(i+1)>>1]);
		puts("");++md;
		if(md==ans) exit(0);return;
	}
	for(int i=0;i<g[x].size();i++){
		v[g[x][i]]=1;dfs(x+1);v[g[x][i]]=0;
	}
}

int main(){
	read(n);read(m);read(k);
	for(int i=1,x,y;i<=m;i++){
		read(x);read(y);
		add(x,y);add(y,x);
	}
	dijkstra();build();
	printf("%d\n",ans);
	dfs(2);
	return 0;
}