SSL1114 和为零 (DFS)

---

原题链接

外网进不去

题目大意

在$1\sim n$各个数字中间放$“+”$或$“-”$或不放符号组成一个算式，求得数为$0$的方案
$\mathbf{S}\mathbf{a}\mathbf{m}\mathbf{p}\mathbf{l}\mathbf{e}$ $\mathbf{I}\mathbf{n}\mathbf{p}\mathbf{u}\mathbf{t}$

7

$\mathbf{S}\mathbf{a}\mathbf{m}\mathbf{p}\mathbf{l}\mathbf{e}$ $\mathbf{O}\mathbf{u}\mathbf{t}\mathbf{p}\mathbf{u}\mathbf{t}$

1+2-3+4-5-6+7
1+2-3-4+5+6-7
1-2 3+4+5+6+7
1-2 3-4 5+6 7
1-2+3+4-5+6-7
1-2-3-4-5+6+7

解题思路

$dfs$所有的符号可能，然后再通过判断得数是否为$0$就可以了
注意这里的优先顺序是不放符号$>“+”>“-”$
即使当前没有符号也要输出一个空格

上代码

#include<iostream>
using namespace std;

int n;
char op[10];

bool pd()
{
    int sum=0,num=1;
    char oper='+';
    for(int i=1; i<n; i++)
    {
        if(op[i]==' ')
            num=num*10+(i+1);
        else
        {
            if(oper=='+') sum+=num;
            else sum-=num;
            oper=op[i];
            num=(i+1);
        }
        // cout<<op[i]<<" "<<sum<<" "<<num<<endl;
    }
    if(oper=='+') sum+=num;
    else sum-=num;
    return sum==0 ;
}

void print()
{
    cout<<"1";
    for(int i=1; i<n; i++)
        cout<<op[i]<<i+1;
    cout<<endl;
}

void dfs(int k)
{
    if(k==n)
    {
        if(pd()) print();
        return;
    }
    op[k]=' ';
    dfs(k+1);
    op[k]='+';
    dfs(k+1);
    op[k]='-';
    dfs(k+1);
    op[k]=char(0);
}

int main()
{
    cin>>n;
    dfs(1);
    return 0;
}

完美切题$\sim$