计蒜客 组合运算式

题目如下：

请考虑一个被空格分隔的，由1到N的整数组成的递增数列：1 2 3 … N。现在请在数列中插入表示加的“+”，或者表示减“-”，亦或者表示空白的“ ”(例如1-2 3就等于1-23)，来将每一对数字组合成一个表达式（第一个数字前无空格）。计算该表达式的结果并判断其值是否为0。请你写一个程序找出所有产生和为零的长度为N的数列。

输入为一行，包含一个整数N（3≤N≤9）。

输出为所有在每对数字间插入“+”, “-”, 或 “ ”后能得到和为零的数列，并按照字典（ASCII码）序排列。

样例输入

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
1-2-3-4-5+6+7

思路如下：

• 本题的基本思路就是深度优先搜索，突破口在于如何处理好三种符号的插入以及求和；

• 我用一个数组来存储数列，用0、1、2分别表示 空格 加号 减号；

• 首先是将符号插入对应位置，插入顺序是 空格 加号 减号（即这三种符号的字典序），以这个顺序插入符号，最后输出的结果自然就是字典序排列的；
• 接下来就是求和判断是否满足条件，从第一个符号位开始判断，若为“+”，则将后一位数字压栈，若为“-”，则将下一位数字取相反数压栈，接下来是最难处理的空格，若为空格，则取出栈顶元素，判断正负，与当前符号位的后一位数字结合（正加负减），将结果入栈；
• 接下来就是将栈内元素依次出栈求和（记得每次求和之后将标记置为0）进行判断，满足则输出（用数组存储序列的优势就体现出来了，稍加判断输出即可）。

附上代码

#include<iostream>
#include<stack>
using namespace std;

int N;
int sum = 0;
int X[20]={0};
int book[20]={0};
stack<int> res;

void dfs(int step)
{
    //符号插入完毕 
    if(step==N)
    {
        //先将 1 入栈 
        res.push(X[0]);
        //根据当前符号判断入栈 
        for(int i=1;i<2*N-1;i=i+2)
        {
            // 1 --- 空格  
            if(X[i]==1)
                res.push(X[i+1]);
            // 2 --- 加号 
            else if(X[i]==2)
                res.push(-X[i+1]);
            // 3 --- 减号 
            else
            {
                int e = res.top();
                res.pop();
                if(e<0)
                    res.push(e*10-X[i+1]);
                else
                    res.push(e*10+X[i+1]);
            }
        }
        //求和 
        while(!res.empty())
        {
            sum+=res.top();
            res.pop();
        }
        //求和结果为 0 则输出 
        if(sum==0)
        {
            for(int i=0;i<2*N-3;i+=2)
            {
                cout<<X[i];
                switch(X[i+1])
                {
                    case 0:
                        cout<<' ';
                        break;
                    case 1:
                        cout<<'+';
                        break;
                    case 2:
                        cout<<'-';
                        break;
                }
            }
            cout<<X[2*N-2]<<endl;
        }
        //很重要，每找出一个序列，将和置为 0 
        sum=0;
        return;
    }

    //按照 0 1 2 ，即 空格 加 减 的顺序插入 
    for(int i=0;i<3;i++)
    {
        if(book[step*2-1]==0)
        {
            X[step*2-1]=i;
            book[step*2-1]=1;
            dfs(step+1);
            book[step*2-1]=0;
        }
    }

    return;
}

int main()
{
    cin>>N;
    //首先将数字插入序列 
    for(int i=0;i<N;i++)
        X[2*i]=i+1;
    dfs(1);

    return 0;
}

初衷不变，仅供参考学习，交流讨论，不喜勿喷，有更优解法欢迎评论

附上上一篇博客地址：
计蒜客 强迫症的吃货