洛谷P2404 自然数的拆分问题(回溯)
#洛谷P2404 自然数的拆分问题(回溯)#
参考题解:https://www.luogu.com.cn/blog/CM0/solution-p2404
(第一次写博客,主要是给自己看的,记录一下第一次接触回溯法自己的理解过程,文章可能存在许多小错误,敬请指正;D)
题目描述
任何一个大于1的自然数n,总可以拆分成若干个小于n的自然数之和。现在给你一个自然数n,要求你求出n的拆分成一些数字的和。每个拆分后的序列中的数字从小到大排序。然后你需要输出这些序列,其中字典序小的序列需要优先输出。
输入格式
输入:待拆分的自然数n。
输出格式
输出:若干数的加法式子。
输入输出样例
输入#1
7
输出#1
1+1+1+1+1+1+1
1+1+1+1+1+2
1+1+1+1+3
1+1+1+2+2
1+1+1+4
1+1+2+3
1+1+5
1+2+2+2
1+2+4
1+3+3
1+6
2+2+3
2+5
3+4
代码
建立一个数组记录每次分解的情况,利用递归调用的特性从后往前回溯穷举每一种可能。
#include<iostream>
#include<cstdio>
using namespace std;
int n;
int a[100009]={1};
void output(int t)
{
for(int i=1;i<=t-1;i++)
cout<<a[i]<<"+";
cout<<a[t]<<endl;
}
void dfs(int s,int t)
{
for(int i=a[t-1];i<=s;i++)
{
if(i<n)
{
a[t]=i;
s-=i;
if(s==0)output(t);
else dfs(s,t+1);
s+=i;
}
}
}
int main()
{
cin>>n;
if(n==1)cout<<"1"<<endl;
else
dfs(n,1);
return 0;
}
↓刚开始不明白在dfs前修改的s改回去有什么用↓
if(i<n) {
// cout<<" 循环次数 "<<++o<<endl;
a[t]=i;
s-=i;
//cout<<t<<" "<<s<<" "<<i<<endl;
if(s==0) output(t);
else dfs(s,t+1);
// cout<<"深搜 "<<t<<" "<<s<<" "<<i<<endl;
s+=i;
//cout<<"回溯 "<<t<<" "<<s<<" "<<i<<endl;
}
于是运行n=3,模拟程序执行过程
发现程序在输出1+1+1后,深搜和回溯的t,从3变成了2,且没有进入for循环,想到书上曾讲到函数的递归调用类似于栈,于是画图模拟
递归调用(入栈)↓
结束调用(出栈)↓
然后就懂了。(懒了不想写了 (>人<;)