【递归+二叉树思想+搜索】 Alice and the Cake题解

Alice and the Cake题解

AC记录：记录-洛谷

题面翻译（大概就是题目大意）

执行恰好 $n-1$ 次操作，每次操作可以选择当前所有蛋糕中满足其重量 $w⩾2$ 的一块，然后将其分为质量分别为 $\lfloor \frac{w}{2}\rfloor$ 和 $\lceil \frac{w}{2}\rceil$ 的两块。

现在，给定执行完所有操作后 $n$ 块蛋糕的重量是 $a_1,a_2\cdots ,a_n$，请判断是否存在一个初始蛋糕重量和操作序列使得最终能得到给定的 $n$ 块蛋糕。如果能，输出 YES，否则输出 NO。

数据范围：

• $t$ 组数据，$1⩽t⩽10^4$。
• $1⩽n,\sum n⩽2\times 10^5$。
• $1⩽a_i⩽10^9$。

Translated by Eason_AC

大致思路

分别递归+搜索 查看左半边蛋糕（左子树）和右半边蛋糕（右子树）是否符合规则。（二叉树+搜索+递归）

小小的代码

solve函数：

void solve() {
	p.clear();
	long long int n;
	cin >> n;
	long long int x;
	long long int sum = 0;
	for(int i = 1; i <= n; i++) {
		cin >> x;
		p[x]++;
		sum += x;
	}
	if(dfs(sum))
		cout << "YES" << endl;
	else
		cout << "NO" << endl;
}

递归

int dfs(long long int s) {
	if(!s)
		return 0;
	if(mp[s]) {
		mp[s]--;
		return 1;
	}
	return dfs(s / 2) && dfs(s - (s >> 1));
}

---

寂寞的人唱伤心的歌~