社区讨论

全部数据都TLE了,求优化

P3799小 Y 拼木棒参与者 1已保存回复 0

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
0 条
当前快照
1 份
快照标识符
@mhj350r7
此快照首次捕获于
2025/11/03 19:57
4 个月前
此快照最后确认于
2025/11/03 19:57
4 个月前
查看原帖
CPP
#include <iostream>
using namespace std;
const int mod = 1e9 + 7;
int n, ans = 0;
int a[100005];
void dfs(int cnt, int x, int y, int z, int t, int id) {
	if(id > n) return;
	if(cnt == 4) {
		if(x == 0 || y == 0 || z == 0 || t == 0) return;
		else {
			if(x + y == z && x + y == t) {
				ans++;
				return;
			}
		}
	}
	dfs(cnt + 1, x + a[id], y, z, t, id + 1);
	dfs(cnt + 1, x, y + a[id], z, t, id + 1);
	dfs(cnt + 1, x, y, z + a[id], t, id + 1);
	dfs(cnt + 1, x, y, z, t + a[id], id + 1);
	dfs(cnt, x, y, z, t, id + 1);
}
int main() {
	cin >> n;
	for(int i = 0; i < n; i++) {
		cin >> a[i];
	}
	dfs(0, 0, 0, 0, 0, 0);
	ans /= 4;		// x+y=z=t, x+y=t=z, y+x=z=t, y+x=t=z
	cout << ans % mod << endl;
	return 0;
}

回复

0 条回复,欢迎继续交流。

正在加载回复...