USACO2.3.3 Zero Sum 和为零 解题报告(dfs)

Description

请考虑一个由1到N（N=3, 4, 5 ... 9）的数字组成的递增数列：1 2 3 ... N。 现在请在数列中插入“+”表示加，或者“-”表示减，抑或是“ ”表示空白，来将每一对数字组合在一起（请不在第一个数字前插入符号）。 计算该表达式的结果并注意你是否得到了和为零。 请你写一个程序找出所有产生和为零的长度为N的数列。

Input

单独的一行表示整数N (3 <= N <= 9)。

Output

按照ASCII码的顺序，输出所有在每对数字间插入“+”, “-”, 或 “ ”后能得到和为零的数列。(注意:就算两个数字之间没有插入符号也应该保留空格)

Sample Input

7

Sample Output

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,如果数字x前面的符号是'+',则sum+=x,如果为'-',则sum-=x,如果为' ',则将该数字加上(之前的数字*10),sum不变,这样dfs到最底层的时候判断sum是否为0即可

 

#include <map>
#include <cstdlib>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <iostream>
#include <sstream>
#include <algorithm>
#define lowbit(a) (a&(-a))
#define _mid(a,b) ((a+b)/2)
#define debu(a) cout << a <<" ";
#define debug(a) cout << a << endl;
#define _mem(a,b) memset(a,0,(b+3)<<2)
#define fori(a) for(int i=0;i<a;i++)
#define forj(a) for(int j=0;j<a;j++)
#define ifor(a) for(int i=1;i<=a;i++)
#define jfor(a) for(int j=1;j<=a;j++)
#define mem(a,b) memset(a,b,sizeof(a))
#define IN freopen("in.txt","r",stdin)
#define OUT freopen("out.txt","w",stdout)
 
#define IO do{\
    ios::sync_with_stdio(false);\
    cin.tie(0);\
    cout.tie(0);}while(0)
#define mp(a,b) make_pair(a,b);
using namespace std;
typedef long long ll;
const int maxn =  60;
const int INF = 0x3f3f3f3f;
const int inf = 0x3f;
const double EPS = 1e-7;
const double Pi = acos(-1);
const int MOD = 1e9+7;
char res[maxn];
string ans[maxn];
int cn;
int cnt;
int n;
//k表示当前遍历到第几个数字,sum代表前K层和,fro代表上一层数字
void dfs(int k,int sum,int fro){
    if(k-1==n){
        sum += fro;
        if(sum==0)
            ans[cn++] = res;
        return;
    }
    res[cnt++] = '+';
    dfs(k+1,sum+fro,k);      //如果当前数字前的符号为'+',说明下一层的前一个数字为K
    res[cnt-1] = '-';
    dfs(k+1,sum+fro,-k);    //同理
    res[cnt-1] = ' ';
    dfs(k+1,sum,(abs(fro)*10+k)*(fro>=0?1:-1));//如果为空格,则需要加上上一层数字*当前层数字,比如上上层的-2传到了下一层
//但是上一层符号为' '那么这时候上一层数字传下来的数字应该为(abs(-2)*10+3)*-1 = -23
//保证符号和数字的正确性
    res[cnt--] = 0;
}
 
int main() {
    cin >> n;
    dfs(2,0,1);
    sort(ans,ans+cn);
    fori(cn){
        forj(ans[i].size())
            cout << j+1 << ans[i][j];
        cout << n << endl;
    }
    return 0;
}
 

 

转载于:https://www.cnblogs.com/bestsort/p/10588844.html