P1473 [USACO2.3] 零的数列 Zero Sum

✅博客主页:爆打维c-CSDN博客​​​​​​ 🐾

🔹分享自己学习AI/编程知识的过程 🐾

🔹我的GitHub代码仓库  https://github.com/lyy-0118

一、题目链接

刷题平台为洛谷，链接如下：

P1473 [USACO2.3] 零的数列 Zero Sum - 洛谷

二、题目分析

1.题目描述

2.分析题意

对应算法知识，本题主要考察：

• DFS / 回溯枚举
• 表达式求值
• 字符串构造
• 字典序 / ASCII 顺序输出

给定数字序列 1,2,3,...,N，要求在相邻数字之间插入三种符号：空格、+、-。

空格表示把两个数字拼接成一个数，例如 1-2 3 等价于 1-23。

需要找出所有最终表达式结果为 0 的方案，并按 ASCII 顺序输出。

因为 ASCII 顺序是：空格 < + < -，所以搜索时符号顺序也应为：' '、'+'、'-'。

数据范围很小，3 <= N <= 9，最多只有 3^(N-1) 种情况，可以直接枚举。

由于每两个相邻数字之间有 3 种选择，所以可以用 DFS 枚举所有符号组合。

枚举完成后，再计算表达式的值。

计算时要注意空格代表数字拼接，例如 2 3 应视为 23，而不是两个独立数字。

因为 N <= 9，暴力搜索完全可行。

3.解题思路

1. 用dfs枚举两个相邻数字的符号，依次尝试：空格、+、-;用一个字符数组存下当前的符号排列情况；
2. 当枚举到n-1的位置，判断当前符号排列情况下，此数列的sum是否为0：从数字1开始，维护当前数字num和前一个符号last_op；如果当前op[i]为空格，则把后一个数字拼接到num后；如果当前op[i]为‘+’或‘-’，不需要拼接，则按上一个符号last_op把num加入sum中，再更新新的op和num。若结果为 0，输出表达式。

三、实现代码，带简单注释

#include <bits/stdc++.h>
using namespace std;
int N;
char op[10];  // op[i] 表示数字 i 和 i+1 之间的符号

// 计算当前表达式的值
int calc(){
	int sum=0;
	int num=1;   //正在处理的数字,从1开始
	int last_op=op[0];
	for(int i=1;i<N;i++){ //处理符号
		int nextNum=i+1;
		if(op[i]==' '){ //符号为空格，先拼接，下一步再加减
			num=num*10+nextNum;
		}else{          //符号为 + -
			if(last_op=='+') sum+=num;
			else sum-=num;
			
			// 更新符号和当前数字
			last_op=op[i];
			num=nextNum;
		}
	}
	// 处理最后一个数字
    if (last_op == '+') sum += num;
    else sum -= num;
    return sum;
}

// 输出当前表达式
void printExpr() {
    for (int i = 1; i <= N; i++) {
        cout << i;
        if (i < N) cout << op[i];
    }
    cout << '\n';
}

//DFS枚举符号
void dfs(int pos){
	if(pos==N){
		if(calc()==0){
			printExpr();
		}
		return ;
	}
	// 按 ASCII 顺序：空格、+、-
	op[pos]=' ';
	dfs(pos+1);
	op[pos]='+';
	dfs(pos+1);
	op[pos]='-';
	dfs(pos+1);
}

int main() {
    cin >> N;
    op[0]='+';
    dfs(1);
    return 0;
}