F. Graph Without Long Directed Paths

Problem - 1144F - Codeforces

思路

       要求每条路径长度都必须小于2，才输出YES。因此保持相连点的入度出度的交替，即当前点只有入度那么下个点就只有出度。

        YES时的输出问题，开始时记录u,v。并创建vis记录哪个点入度，那个点出度。输出答案时遍历u输出该点入度还是出度即可。

实现

#include<bits/stdc++.h>
using namespace std;
const int maxn = 3e5;
struct Edge{int nxt,to,id;}edge[maxn<<1];
int head[maxn],cnt;
void add(int u,int v,int id){
	edge[++cnt].nxt = head[u];
	edge[cnt].to = v;
	edge[cnt].id = id;
	head[u] = cnt;
}
int n,m,e[maxn][2],vis[maxn],f;

void dfs(int now,int orin){
	if (f) return;
	for(int i=head[now];i;i=edge[i].nxt){
		int to = edge[i].to,id = edge[i].id;
		if (vis[to]!=-1){
			if (vis[to]==orin) return void(f=1);
		}else{
			vis[to] = orin^1;
			dfs(to,orin^1);
		}
	}
}

void solve(){
	cin>>n>>m;
	f = 0;
	for (int i=1;i<=n;i++) vis[i]=-1;
	for(int i=1;i<=m;i++){
		cin>>e[i][0]>>e[i][1];
		add(e[i][0],e[i][1],i);add(e[i][1],e[i][0],i);
	}
	dfs(1,0);
	if (f) cout<<"NO"<<endl;
	else{
		cout<<"YES"<<endl;
		for(int i=1;i<=m;i++){
			if (vis[e[i][0]]) cout<<0;
			else cout<<1;
		}cout<<endl;
	}
}

int main(){
	ios::sync_with_stdio(0);cin.tie(0);
	solve();
}

感悟

        链式前向星的edge数组开大小时，要注意边的数量