http://202.197.224.59/OnlineJudge2/index.php/Problem/read/id/1241
很好的一道dp题!
题意:给出n,m,问存在多少种方案,满足a[1] + a[2] + .. + a[m] = n,且a[1] < a[2] < … < a[m]
思路:因次数是递增的,且全是正数,因此m * (m + 1) / 2 <= n,一旦不满足不等式,结果必为0.同时,在这条式子的限制下,m不会大于446
将a[i] - (i - 1),则每个数变成了非递减,即a[1] <= a[2] <= … <= a[m]
用dp[i][j]表示前i个数和为j的方案数,则dp[i][j] = dp[i][j - i] + dp[i - 1][j - 1],意思是
1)dp[i][j] += dp[i][j - i],即给每个数加1,
为什么是每个数都减i-1呢,因为状态更新的时候要从i-1往上更新,也就是说最低位应从1开始,那么如果减i的话,每一位就都从0开始了!!
那么转移方程式就是根据最低位是不是1来讨论的,如果不是1的话就是,之前的每一位都加1,如果是1的话就是在开头填个1!再这两种情况已经包含了所有,其余填2,填几都可以用填1代替!
很巧妙的转换,也是套路!
注意空间,应该用滚动数组,而且在处理多组数据的时候也有技巧!先把询问都记录下来,然后在滚动到相应位置时候输出!orz!
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
vector<int> val[510];
vector<int> id[510];
const int N = 100010;
int dp[2][N],ans[N * 10];
const int mod = 1000000007;
//dp[i][j] = dp[i][j - i] + dp[i - 1][j - 1]
int main()
{
int t;
scanf("%d",&t);
for(int i = 1; i <= t; i ++) {
int n,m;
scanf("%d%d",&n,&m);
if(m > 500 || m * (m + 1) / 2 > n) continue;
val[m].push_back(n - m * (m - 1) / 2);
id[m].push_back(i);
}
dp[0][0] = 1;
for(int i = 1; i <= 500; i ++) {
int flag = i & 1;
for(int j = 0; j < N; j ++) dp[flag][j] = 0;
for(int j = i; j < N; j ++) {
dp[flag][j] = dp[flag ^ 1][j - 1] + dp[flag][j - i];
if(dp[flag][j] >= mod) dp[flag][j] -= mod;
}
for(int j = 0; j < val[i].size(); j ++) {
ans[id[i][j]] = dp[flag][val[i][j]];
//cout << i << " " << id[i][j] << " " << val[i][j] << " " << dp[flag][val[i][j]] << " " << dp[1][3] << endl;
}
}
for(int i = 1; i <= t; i ++) printf("%d\n",ans[i]);
return 0;
}