题目:
假如一个长度为 n的正整数序列满足所有前缀和 都是平方数,那么称这种序列为前缀平方序列。
条件1<=si<=x 取模1e9+7
首先找出小于x的平方数有几个。
然后用二项式定理 算出小于x的平方数中取n个的种数。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX=1e3+10;
const ll mod=1e9+7;
int binomial(int n, int k)
{
int C[MAX][MAX];
for(int i=0;i<=n;i++)
{
C[i][0]=1;
C[i][i]=1;
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<i;j++)
{
C[i][j]=C[i-1][j-1]+C[i-1][j];
C[i][j]%=mod;
}
}
return C[n][k];
}
int main(){
int n,x;cin>>n>>x;
//1 4 9 16
//1 4 9 25
//1 4 16 25
//1 9 16 25
//4 9 16 25
cout<<binomial(sqrt(x),n)<<endl;
}