BZOJ 3360 [Usaco2004 Jan]算二十四 dfs+模拟

Description

    写一个程序，给出D(2≤D≤10)个数字，按原顺序在数字间加+，一，×算出24，且不使用括
号．优先级按正常的优先级处理，即先做乘法后做加减法．输出有多少种不同的方案数．

Input

    第1行：一个整数D.
    第2到D+1行：D个整数，在1到50之间．

Output

 
    输出方案总数．

Sample Input

5

6

4

2

8

16

Sample Output

4

样例说明

四种方法分别是6x4x2-8-16，6-4- 2+8+16，6x4-2 x8+16，6×4+2×8-16.

HINT

传送门
总共才9个符号啊= =
O(3^D)爆搜就好了。。
中间可能要用到+,-,*表达式求值，那就先把-后面的数取负了，然后再*，最后+。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int n,ans,g[11],a[11];
ll b[11];
ll calc(){
for (int i=1;i<=n;i++) b[i]=a[i];
for (int i=1;i<n;i++)
if (g[i]==2) b[i+1]=-b[i+1];
for (int i=1;i<n;i++)
if (g[i]==0) b[i+1]=b[i+1]*b[i],b[i]=0LL;
ll s=0LL;
for (int i=1;i<=n;i++) s+=b[i];
return s;
}
void dfs(int now){
if (now==n){
if (calc()==24) ans++;
return;
}
g[now]=0,dfs(now+1);
g[now]=1,dfs(now+1);
g[now]=2,dfs(now+1);
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
dfs(1);
printf("%d\n",ans);
return 0;
}