[Daimayuan] 农田划分（C++，数学，BFS）

题目描述

约翰是一个农场主，他的农场有n块田，编号从 $1$到 $n$，这 $n$块田通过 $m$条双向道路相连(数据保证这$n$块田都是联通的)，我们假设第$i$块田会产生 $2^{i}kg$ 的收益，现在约翰想把农田的工作全部交给自己的两个孩子，划分方式必须满足以下规则:

1.每一块田都需要恰好被分给一个孩子.

2.分给两个孩子的农田必须是联通的.就是说对于任意一个孩子在划分给自己的任意一块田，都可以不经过另外一个孩子的田，到达自己的任意一块田.

3.划分给两个孩子的收益必须尽可能的相等，如果无法相等，年长的孩子会得到大的那一份.

对于第 $i$块田，如果你要把它分给年长的孩子，请输出A,否则输出B.

题目输入

第一行输入两个整数分别代表 $n,m$ 接下来 $m$行，每个两个整数$u,v$，代表这两块农田通过一条双向道路直接相连，数据保证没有重边和自环

题目输出

输出一个字符串，代表答案

样例输入1

3 2
1 3
3 2

样例输出1

ABA

样例输入2

6 6
3 5
2 6
1 3
3 6
5 1
4 6

样例输出2

BABABA

数据范围

$2\le n\le 3e5,1\le m\le 3e5$

解题思路

首先，我们要知道：

对于数列$[1,2,2^2,...,2^{n-1},2^n]$，$2^n$是大于之前所有的数加和的。

利用这个性质，因为我们要保证年长的孩子分得的土地更多，所以$n$号田地一定会被分给年长的孩子。

继续利用这个性质，因为我们又要保证划分尽可能的相等，所以$n-1$号田地一定会被划分给另外一个孩子。

这有什么用呢？

我们的具体算法是这样的：

删除$n$号田地及与其相连的所有通路，然后从$n-1$号田地开始$BFS$，所有被搜索到的田地均分给另外一个孩子，剩下的田地归年长的孩子。

算法时间复杂度为$O(n)$。

最后，AC代码如下：

#include <iostream>
#include <queue>
#include <string.h>
using namespace std;
const int max_n = 3e5;
const int max_m = 3e5;

struct edge { int v, next; }edges[max_m * 2];
int tot = -1, heads[max_n + 1];
queue<int>q;
bool book[max_n + 1];

void add_edge(int u, int v) {
	edges[++tot] = { v,heads[u] }; heads[u] = tot;
	edges[++tot] = { u,heads[v] }; heads[v] = tot;
}

void bfs(int s) {
	q.push(s);//初始化

	//bfs主体
	while (!q.empty()) {
		int head = q.front();
		q.pop();//取出首节点

		if (book[head]) continue;
		book[head] = true;//访问判断

		for (int i = heads[head]; i != -1; i = edges[i].next) {//进行尝试
			int v = edges[i].v;
			if (book[v]) continue;//重复访问
			q.push(v);
		}
	}
}

int main() {
	int n, m, u, v;
	cin >> n >> m;
	memset(heads + 1, -1, sizeof(int) * n);
	for (int i = 0; i < m; i++) {
		cin >> u >> v;
		if (u == n || v == n) continue;//逻辑删除
		add_edge(u, v);
	}
	bfs(n - 1);
	for (int i = 1; i <= n; i++) {
		if (book[i]) cout << 'B';
		else cout << 'A';
	}
	return 0;
}