题目地址
可以知道答案的上限是36,我们可以先处理出所有的方案数。
然后dfs,对每个状态,记录一下当前的答案,dfs要搜一下要不要取当前这个方案。
剪枝的话就是如果当前的答案加上剩下所有方案数小于当前最优解的时候就不用dfs下去了。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int num[10];
struct node{
int i, j, k;
};
vector<node>ve;
void init() {
for(int i = 1; i <= 9; i++) {
for(int j = 1; j+i <= 9; j++) {
ve.push_back(node{i, j, i+j});
}
}
}
int ans;
void dfs(int p, int res) {
if(p == ve.size()) {
ans = max(ans, res);
return;
}
if(ve.size()-p+res <= ans) return;
if(num[ve[p].i] >= 1 && num[ve[p].j] >= 1 && num[ve[p].k] >= 1) {
num[ve[p].i]--;num[ve[p].j]--;num[ve[p].k]--;
if(num[ve[p].i]>=0 && num[ve[p].j]>=0 && num[ve[p].k]>=0) dfs(p+1, res+1);
num[ve[p].i]++;num[ve[p].j]++;num[ve[p].k]++;
}
dfs(p+1, res);
}
int T = 0;
void solve() {
ans = 0;
for(int i = 1; i <= 9; i++) {
scanf("%d", &num[i]);
}
dfs(0, 0);
printf("Case #%d: %d\n", ++T, ans);
}
int main() {
int Case;
init();
scanf("%d", &Case);
while(Case--)
solve();
return 0;
}