牛客寒假算法基础集训营1C-红和蓝 题解

题目大意

你拿到了一棵树，请你给每个顶点染成红色或蓝色。要求每个红点相邻的点有且仅有一个红点，每个蓝点相邻的点有且仅有一个蓝点。

解题思路

这个题其实做法很暴力，就是看你能不能想到了。

想通了就很简单。

首先dfs一次这棵树。

从叶子节点开始，每个点与他的父节点认定为一组。

如果某个点的父节点是被标记了，那就肯定不行。

标记完之后，再dfs一遍。

如果是一组的就相同的颜色，否则就相反。

完整代码

#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <queue>
#include <deque>
#include <cstring>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <list>
#include <string>
#include <vector>
#include <algorithm>
#include <sstream>
#include <unordered_map>
using namespace std;
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=a;i>=n;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define pll pair<ll,ll>
#define pii pair<int,int>
#define bg begin
#define rbg rbegin
#define ed end
#define endl '\n'
#define dbg(x) cout << #x << "===" << x << endl
typedef double db;
typedef long long ll;
typedef unsigned long long ull;

const int N = 1e5 + 5;
int flag = 1, cnt = 0;
int head[N], gp[N], ans[N];
struct {
	int t, next;
}edge[N<<1];

void build(int u, int w) {
	edge[++cnt].t = w, edge[cnt].next = head[u];
	head[u] = cnt;
}

void dfs1(int x,int fa) {
	for (int i = head[x]; ~i; i = edge[i].next) {
		int t = edge[i].t;
		if (t == fa) continue;
		dfs1(t, x);
	}
	if (!gp[x]) {
		if (gp[fa]) {
			flag = 0; return;
		}
		gp[x] = gp[fa] = ++cnt;
	}
}

void dfs2(int x, int fa) {
	for (int i = head[x]; ~i; i = edge[i].next) {
		int t = edge[i].t;
		if (t == fa) continue;
		if (gp[x] == gp[t])	ans[t] = ans[x];
		else ans[t] = ans[x] ^ 1;
		dfs2(t, x);
	}
}

int gcd(int a, int b) {
	return !b ? a : gcd(b, a % b);
}

ll qpow(ll a, ll b, ll mod) {
	ll ans = 1;
	while (b) {
		if (b & 1)	ans = (ans * a) % mod;
		a = (a * a) % mod, b >>= 1;
	}
	return ans % mod;
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	//freopen("D:\\测试用\\in.txt", "r", stdin);
	//freopen("D:\\测试用\\out.txt", "w+", stdout);
	int n; cin >> n;
	memset(head, -1, sizeof(head));
	rep(i, 2, n) {
		int u, v; cin >> u >> v;
		build(u, v), build(v, u);
	}
	gp[0] = 1;
	dfs1(1, 0);
	if (!flag) {
		cout << -1 << endl;
		return 0;
	}
	dfs2(1, 0);
	rep(i, 1, n) {
		if (ans[i])	cout << 'R';
		else cout << 'B';
	}
	cout << endl;
	return 0;
}