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

                                                                           小明的数学作业（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

Sample Output

#include 
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");}
}

HINT

Source

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！