洛谷 P1025 & [NOIP2001提高组] 数的划分

题目链接

https://www.luogu.org/problemnew/show/P1025

解题思路

一道简单的dfs题，但是需要剪枝，否则会TLE。

---

我们用dfs(a,u,num)来表示上一个数为a，已经搜索完了a个数，现在的和是num。

 1 #include<iostream>
 2 using namespace std;
 3 int n,k,a;
 4 long long ans;
 5 void dfs(int a,int u,int now){
 6     if(now>n) return;
 7     if(u>k) return; 
 8     if(u==k){
 9         if(now==n) ans++;
10         return;
11     }
12     for(int i=a;i<=n;i++){
13         dfs(i,u+1,now+i);
14     }
15 }
16 int main(){
17     cin>>n>>k;
18     dfs(1,0,0);
19     cout<<ans;
20     return 0;
21 }

未剪枝代码

---

剪枝：现在搜索完u层了，那么还剩下k-u层，因为后一个数一定大于等于前一个数，所以i从a开始，而且后边的数一定>=i，所以i*(k-u)表示后边的数的最小和，如果现在的和加上后面的最小和已经大于n了，直接不去继续递归了。

最终代码

 1 #include<iostream>
 2 using namespace std;
 3 int n,k,a;
 4 long long ans;
 5 void dfs(int a,int u,int now){
 6     if(now>n) return;//边界 
 7     if(u>k) return; 
 8     if(u==k){
 9         if(now==n) ans++;
10         return;
11     }
12     for(int i=a;now+i*(k-u)<=n;i++){//剪枝 
13         dfs(i,u+1,now+i);
14     }
15 }
16 int main(){
17     cin>>n>>k;
18     dfs(1,0,0);
19     cout<<ans;
20     return 0;
21 }

//NOIP2001提高组 t2