您当前的位置: 首页 >  c++

征战蓝桥 —— 2014年第五届 —— C/C++A组第10题——波动数列

发布时间:2019-03-16 23:33:36 ,浏览量:0

题目
观察这个数列:
1 3 0 2 -1 1 -2 ...

这个数列中后一项总是比前一项增加2或者减少3。

栋栋对这种数列很好奇,他想知道长度为 n 和为 s 而且后一项总是比前一项增加a或者减少b的整数数列可能有多少种呢?
【数据格式】
输入的第一行包含四个整数 n s a b,含义如前面说述。
输出一行,包含一个整数,表示满足条件的方案数。由于这个数很大,请输出方案数除以100000007的余数。

例如,输入: 4 10 2 3 程序应该输出: 2

【样例说明】 这两个数列分别是2 4 1 3和7 4 1 -2。

【数据规模与约定】

对于10%的数据,1<=n<=5,0<=s<=5,1<=a,b<=5; 对于30%的数据,1<=n<=30,0<=s<=30,1<=a,b<=30; 对于50%的数据,1<=n<=50,0<=s<=50,1<=a,b<=50; 对于70%的数据,1<=n<=100,0<=s<=500,1<=a, b<=50; 对于100%的数据,1<=n<=1000,-1,000,000,000<=s<=1,000,000,000,1<=a, b<=1,000,000。

资源约定: 峰值内存消耗 < 256M CPU消耗 < 1000ms

请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。

所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。

注意: main函数需要返回0 注意: 只使用ANSI C/ANSI C++ 标准,不要调用依赖于编译环境或操作系统的特殊函数。 注意: 所有依赖的函数必须明确地在源文件中 #include , 不能通过工程设置而省略常用头文件。

提交时,注意选择所期望的编译器类型。

代码
#include  #include  #include  #define MOD  100000007 using namespace std; typedef long long LL; int n, s, a, b; long long ans; void printPath(const vector<int> &path); void solve1(); void dp1(); void dp2(); void dp3(); /**
 *
 * @param x 上一项
 * @param cnt 截止上一项已经有多少个
 * @param sum 截止上一项已经求得的和
 */ void dfs(LL x, int cnt, int sum, vector<int> path) { if (cnt == n) { if (sum == s) { ans++; } //        printPath(path); if (ans > MOD)ans %= MOD; return; } path.push_back(x + a); dfs(x + a, cnt + 1, sum + x + a, path); path.erase(path.end() - 1); path.push_back(x - b); dfs(x - b, cnt + 1, sum + x - b, path); } void printPath(const vector<int> &path) { for (int i = 0; i < path.size(); ++i) { cout << path[i] << " "; } cout << endl; } void solve2(); int main(int argc, const char *argv[]) { scanf("%d %d %d %d", &n, &s, &a, &b); //    solve1(); //    solve2(); //    dp1(); //    dp2(); dp3(); return 0; } /*一维数组*/ void dp3() { int t = n * (n - 1) / 2; int dp[t + 1]; memset(dp, 0, sizeof(dp)); dp[0] = 1; for (int i = 1; i < n; ++i) { for (int j = i * (i + 1) / 2; j >= i; --j) {//减少j的枚举 dp[j] = (dp[j] + dp[j - i]) % MOD; } } for (LL ta = 0; ta <= t; ++ta) { LL i = s - ta * a + (t - ta) * b; if (i % n == 0) (ans += dp[ta]) %= MOD; } cout << ans << endl; } /*2行的数组滚动使用*/ void dp2() { int t = n * (n - 1) / 2; int dp[2][t + 1]; memset(dp, 0, sizeof(dp)); dp[0][0] = 1; dp[1][0] = 1; int row = 0; for (int i = 1; i < n; ++i) { row = 1 - row; for (int j = 1; j <= i * (i + 1) / 2; ++j) {//这里可以减少枚举 if (i > j) dp[row][j] = dp[1 - row][j] % MOD; else dp[row][j] = (dp[1 - row][j] + dp[1 - row][j - i]) % MOD; } } //    必须用LL,因为s+tb有可能超出int  10^9+10^6*10^6 for (LL ta = 0; ta <= t; ++ta) { LL i = s - ta * a + (t - ta) * b; if (i % n == 0) (ans += dp[row][ta]) %= MOD;//选择性地累加最后一行 } cout << ans << endl; } void dp1() { int t = n * (n - 1) / 2;//最终式子中a或b的最大的个数(系数) int dp[n][t + 1]; memset(dp, 0, sizeof(dp)); dp[0][0] = 1; for (int i = 0; i < n; ++i) { dp[i][0] = 1; } for (int i = 1; i < n; ++i) { for (int j = 1; j <= t; ++j) { if (i > j) dp[i][j] = dp[i - 1][j] % MOD; else dp[i][j] = (dp[i - 1][j] + dp[i - 1][j - i]) % MOD; } } for (LL ta = 0; ta <= t; ++ta) { LL i = s - ta * a + (t - ta) * b; if (i % n == 0) (ans += dp[n - 1][ta]) %= MOD; } cout << ans << endl; } void solve2() { int t = n * (n - 1) / 2; LL x; LL x1 = (s - a * t) / n; LL x2 = (s + b * t) / n; //    枚举首项 for (x = x1; x <= x2; x++) { //        对x进行初步检测,有的x,无论怎么搭配a,b的数目,都不能得出s for (int ta = 0; ta <= t; ++ta) {//枚举a的数目[0,t] LL cal = x * n + ta * a - (t - ta) * b; if (cal == s) {//减少对x的枚举 vector<int> path; path.push_back(x); dfs(x, 1, x, path); } } } printf("%lli\n", ans); } void solve1() { int t = n * (n - 1) / 2;//最终式子中a或b的最大的个数(系数) LL x; //    枚举首项 LL x1 = (s - a * t) / n; LL x2 = (s + b * t) / n + 1; for (x = x1; x <= x2; x++) { vector<int> path; path.push_back(x); dfs(x, 1, x, path); } printf("%lli\n", ans); } 
关注
打赏
1688896170
查看更多评论

暂无认证

  • 0浏览

    0关注

    109889博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文
立即登录/注册

微信扫码登录

0.4188s