题目链接
题目大意:现在给你N个数,然后问你有多少个区间,对一个区间内的数求和后的到的平均数为K,问有多少个这样的区间,
区间可以理解为子序列,不一定是连续的
思路:这个题目的话,貌似不是第一次做了,之前也有遇到过吧,
dp[i][j] 代表选了i个数时,sum和为j的方案数。
初始化dp[0][0]=1
我是按照反向的思维来的,正向也没问题,
转移方程时dp[i][j]+=dp[i-1][j-x],x为输入的数。
代码:
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=2000;
long long dp[100][500*500+1];
int main()
{
int n,k;
scanf("%d%d",&n,&k);
int sum=0;
dp[0][0]=1;
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
sum+=x;
for(int j=i;j>=1;j--)
{
for(int s=sum;s>=x;s--)
dp[j][s]+=dp[j-1][s-x];
}
}
long long ans=0;
for(int i=1;i<=n;i++)
ans+=dp[i][i*k];
printf("%lld\n",ans);
}