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 100005
using 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]);
}
}