蓝桥杯-波动数列【暴力解法10'】

2019-03-03  本文已影响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

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define MOD  100000007
int n,s,a,b;
LL ans;

void dfs(LL x,int cnt,int sum,vector<int>path)
{
    if(cnt==n)
    {
        if(sum==s)
        {
            ans++;
          //  for (int i=0; i<path.size(); i++)
           // {
         //       cout << path[i] << " ";
          //  }
          //  cout << endl;
        }
        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);
    path.erase(path.end()-1);
}


int main()
{
    scanf("%d%d%d%d",&n,&s,&a,&b);
    int t=n*(n-1)/2;
    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);
    }
    cout<<ans<<endl;
}

上一篇 下一篇

猜你喜欢

热点阅读