社区讨论
全部数据都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 条回复,欢迎继续交流。
正在加载回复...