小明的数学作业【经大神指教】

                                                                           小明的数学作业（dfs +模拟栈头）【经大神指教】

Description

寒假刚刚过去，小明的数学作业还没完成，其中一道题目的意思是给定n个整数，
每两个整数间（包括第一个整数前）都存在一个空位，求是否可以在上面填写四则运算符使得n个数的运算值等于m, 如：1 + 2 * 3 = 7 (除法结果向下取整)，你可以帮助他完成作业吗？

Input

输入数据首先是一个t（t <= 15）,后面跟t组数据，每组数据由两行构成，第一行由两个整数n（n <= 12）和m(m < 2^31)构成，第二行由n个整数a[i] (0 <= i < n && a[i] < 2^31)构成。

Output

如果存在运算值等于m，则输出"YES",否则输出"NO";

Sample Input

23 71 2 33 81 2 3

Sample Output

比赛的时候一直不知道如何 先运算 * / 。。 一直无从下手。。

代码如下：

#include <iostream>
using namespace std;
typedef long long LL;
LL m, a[15];
int n;
int dfs(LL sum, LL head, int i) {
    if (i == n) {
        return sum + head == m;
    }
    return (dfs(sum + head, a[i], i + 1) ||
        dfs(sum + head, -a[i], i + 1) ||
        dfs(sum, a[i] * head, i + 1) ||
        (a[i] != 0 && dfs(sum , head/a[i], i + 1)));
}
int main() {
    int t;
    cin >> t;
    while (t--) {
        cin >> n >> m;
        for (int i = 0; i < n; i++)
            cin >> a[i];
        bool yes = dfs(0, a[0], 1) || dfs(0, -a[0], 1);
        printf("%s\n", yes ? "YES" : "NO");
    }
}

YESNO

HINT

Source