YBTOJ高效进阶递推算法课堂过关T3:数的划分
题面
题目描述
将整数n分成k份,且每份不能为空,任意两个方案不相同(不考虑顺序)。
例如:n=7,k=3,下面三种分法被认为是相同的:
1,1,5; 1,5,1; 1,1,5.
问有多少种不同的分法。
输入格式
两个整数,n和k。
输出格式
输出不同的分法数。
样例
输入样例
7 3
输出样例
4
数据范围
6<n≤200,2≤k≤6。
解法
1、DFS
作为一个蒟蒻,看到“方案数”这个词和数据范围我第一时间想到的是dfs,每次都判断一下当前的数还够不够分
(剩余/2≥上一个数),如果够,那么就从当前到剩余的一半每种都试一下dfs。如果不够,就直接把剩下的所有分成一个数处理。
话不多说直接上核心dfs代码:
void dfs(int p,int x,int sy)
{
if(p>k)
{
ans++;
return ;
}
if(p<k)
for(int i=x;i<=sy/(k-p+1);i++)
{
a[p]=i;
dfs(p+1,i,sy-i);
}
else
{
a[p]=sy;
dfs(p+1,sy,0);
}
}
2、DP
按照上面的解法可以解出这题,主要是因为这题的数据规模小,n最大才200,递归最多也只有6层,如果把n开到1000,k最大100,那么深搜必爆无疑。
如果是这样的话,那么就要另寻解法了,竟然这题能放到递推里面,就一定有递推式或转移方程。
代码:
#include<bits/stdc++.h>
using namespace std;
int n,k,f[1010][20];
int main()
{
cin>>n>>k;
for (int i=1;i<=n;i++)
{
for (int j=1;j<=k;j++)
{
if (i<j)
f[i][j]=0;
else if (i==j)
f[i][j]=1;
else
f[i][j]=f[i-1][j-1]+f[i-j][j];
}
}
cout<<f[n][k];
return 0;
}