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

来源：互联网发布：工业控制网络安全性编辑：程序博客网时间：2024/04/28 06:21

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

0 0