小明的數學作業【經大神指教】

                                                                           小明的數學作業（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