Codeforces 1005 F - Berland and the Shortest Paths

题目：点击打开链接
题意：找出所有点到1点的最短距离的子图，且数量不超过k。

分析：先bfs求出所有点到1点的最短距离，再判断这些点能否改变前驱最短距离保持不变，dfs找到它所能改变的所有前驱，并把连边保存下来，可以自己对照代码手动模拟一下。

代码：

#pragma comment(linker, "/STACK:102400000,102400000")///手动扩栈
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cstdio>
#include<bitset>
#include<vector>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<deque>
#include<list>
#include<set>
#include<map>
using namespace std;
#define debug test
#define pii pair<int, int>
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
#define inf 0x3f3f3f3f
#define eps 1e-10
#define MOD 1000000007
#define PI acos(-1.0)
const int N = 1e6+10;

ll gcd(ll p,ll q){return q==0?p:gcd(q,p%q);}
int to[4][2]={{-1,0},{1,0},{0,-1},{0,1}};

int n,m,k,dis[N],vis[N];
char s[N];
vector<pii> e[N];
vector<int> pre[N];
vector<string> ans;

void bfs() {
    vis[1]=1,dis[1]=0;
    queue<pii> q;
    q.push({1,0});
    pii tp,t;
    while(!q.empty()) {
        tp = q.front(),q.pop();
        for(int i=0;i<e[tp.first].size();i++) {
            int u = e[tp.first][i].first;
            if(!vis[u]) {
                vis[u]=1;
                dis[u]=dis[tp.first]+1;
                q.push(e[tp.first][i]);
            }
        }
    }
}

void dfs(int x) {
    if((int)ans.size()>=k) return ;
    if(x>n) {
        ans.pb(s+1);///char数组转化为string 传入首地址
        return ;
    }
    for(int i=0;i<pre[x].size();i++) {
        int u=pre[x][i];
        s[u]='1';
        dfs(x+1);
        s[u]='0';
    }
}

void sv() {
    for(int i=1;i<=n;i++) {
        for(int j=0;j<e[i].size();j++) {
            pii tp = e[i][j];
            if( dis[tp.first]+1 == dis[i] ) pre[i].pb(tp.second);
        }
    }
    for(int i = 1; i <= m; i++) s[i] = '0';
    dfs(2);
    int sz=min(k,(int)ans.size());
    cout<<sz<<endl;
    for(int i=0;i<sz;i++)
        cout<<ans[i]<<endl;
}

int main() {
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>m>>k;
    for(int i=1;i<=m;i++) {
        int u,v;
        cin>>u>>v;
        e[u].pb({v,i}),e[v].pb({u,i});
    }
    bfs();
    sv();
    return 0;
}

参考博客：http://www.cnblogs.com/widsom/p/9290144.html