Codeforces Round #496 (Div. 3) F. Berland and the Shortest Paths[思维+图论]

题意:一张图有n个顶点,m条边,现在要求选择n-1条边,输出<=k个方案.

每个方案满足:顶点1到任何1个顶点的距离 总和 最小.

思路:每个顶点(除了1),有选择1条边的机会.先bfs求出1到所有顶点的最短距离,假如 dis[u]==dis[v]+1 其中v为相邻顶点,则选这条边,能够最小化答案.那么对于u来说,选择(u,v1) (u,v2) (u,v3)同样能最小化答案.假如问答案有多少种,直接乘起来,这里只要求至多k组,并且k*t<=1e6
 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <queue>
#include <vector>

using namespace std;

int n,m,k;

struct node
{
    int v,id;
};
vector<node> e[200005];
vector<int> c[200005];

int dis[200005];

void bfs()
{
    queue<int> q;
    dis[1]=0;
    q.push(1);

    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        int len=e[u].size();
        for(int i=0;i<len;i++)
        {
            int v=e[u][i].v;
            if(dis[v]==-1)
            {
                dis[v]=dis[u]+1;
                q.push(v);
                c[v].push_back(e[u][i].id);
            }
            else if(dis[v]==(dis[u]+1))
            {
                c[v].push_back(e[u][i].id);
            }
        }
    }
}

int cnt;
int tans[200005];
int tpc[200005];

vector<int> ans[200005];

void dfs(int id)
{
    if(cnt>=k)
        return ;
    int len=c[id].size();
    for(int i=0;i<len;i++)
    {
        int tp=c[id][i];
        tans[id]=tp;
        tpc[tp]=1;
        if(id!=n)
        {
            dfs(id+1);
        }
        else
        {
            for(int j=1;j<=m;j++)
            {
                ans[cnt].push_back(tpc[j]);
            }
            cnt++;
            if(cnt>=k)
                return ;
        }
        tpc[tp]=0;
    }
}
int main() {

    while(~scanf("%d%d%d",&n,&m,&k))
    {
        cnt=0;

        for(int i=0;i<=200000;i++)
        {
            e[i].clear();
            c[i].clear();
            dis[i]=-1;
        }
        for(int i=1;i<=m;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            node cur;cur.id=i;
            cur.v=b;
            e[a].push_back(cur);
            cur.v=a;
            e[b].push_back(cur);
        }

        bfs();
        dfs(2);

        printf("%d\n",cnt);
        for(int i=0;i<cnt;i++)
        {
            for(int j=0;j<m;j++)
            {
                printf("%d",ans[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}