CodeForces 240E E. Road Repairs 树

来源:互联网 发布:vue仿豆瓣app源码下载 编辑:程序博客网 时间:2024/04/30 11:49

CodeForces 240E   E. Road Repairs   树

题意:一个五万个点和五万条边的图,每条边的边权为1或0,问以1为根的最小树形图,并输出方案。

权值为0表示已经修好,权值为1表示需要修理,输出需要休息的边,边号按照给定的顺序从1开始编号;


网上搜的解题报告,有很多是需要联通分量缩点的,但是我从CF上直接扒下来的代码,很短,也没看出需要缩点!

下面这段代码的:

用邻接表存边,然后从根节点开始搜索,不能到达的点,用vis[i]=0来表示,已经修好的路用-1来表示,其他的能到达但是需要修路的话就记录边号!

(注意用vector来实现的时候,只有还没访问到或者访问到但需要重修的路才往里面添加,这样即使里面有回路也不会造成死循环)

最后判断输出既可以了


#include<cstdio>#include<cstring>#include<vector>#include<queue>#define N 100005using namespace std;struct edge{int v,id,sta;}tmp;vector<edge> Adj[N];int n,m,vis[N];int main(){    freopen("input.txt","r",stdin);    freopen("output.txt","w",stdout);    scanf("%d%d",&n,&m);    for (int i=1,u,v,f;i<=m;i++)    {        scanf("%d%d%d",&u,&v,&f);        tmp.v=v,tmp.id=i,tmp.sta=f;        Adj[u].push_back(tmp);    }    queue<int> Q;    Q.push(1),vis[1]=-1;    while(!Q.empty())    {        int p=Q.front();        Q.pop();        for(int i=0;i<Adj[p].size();i++){            int v=Adj[p][i].v,sta=Adj[p][i].sta, id=Adj[p][i].id;            if(!vis[v]||(vis[v]>0&&!sta)) vis[v]=sta?id:-1, Q.push(v);        }    }    vector<int> v;    int flag=0;    for(int i=1;i<=n;i++){        if(vis[i]==0) flag=1;///表示有到达不了的        if(vis[i]!=-1) v.push_back(vis[i]);///能到达但是需要重修的,已经记录了边号了    }    if(flag) puts("-1");    else{        printf("%d\n",v.size());        for(int i=0;i<v.size();i++) printf("%d\n",v[i]);    }}