考虑dp,设precnt为考虑这一位之前的所有非空子序列的数量(即 2^(i-1)-1) 从前往后,对于一个新加入的元素,会对以前的答案产生以下的影响:
首先,对于之前已经产生的答案ans,ans*=2,对于之前所有的序列都可以选择加入该元素也可以选择不加入该元素。
对于该元素加入之前序列的情况,我们要继续讨论该元素的贡献。
我们可以将这种贡献分为两类:
第一类:该元素加入之前的序列产生的贡献
这种贡献我们也可以分两种讨论:
A:
a[i]/k 产生的贡献
ans+=precnt*(a[i]/k)
B:
余项 lf=a[i]-(a[i]/k)*k产生的贡献
为了计算这种贡献,我们需要维护一个h数组,h[j]即(序列数字之和)%k==j的序列的数量
第二类:该元素单独产生的贡献(即序列{a[i]})
类似的分为两类考虑
A:
ans+=(a[i]/k)
B:
h[lf]++
代码如下
#pragma GCC optimize(2)
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<iostream>
#include<queue>
#include<math.h>
#include<string>
#include<map>
#include<functional>
#include<unordered_map>
#include<bitset>
using namespace std;
#define int long long
#define inf 0x3f3f3f3f
#define N 4500
int a[N];
int h[N];
int hnx[N];
int qmi(int m, int k, int p)
{
int res = 1 % p, t = m;
while (k)
{
if (k & 1) res = res * t % p;
t = t * t % p;
k >>= 1;
}
return res;
}
int ans = 0;
signed main(void)
{
cin.tie(0);
cout.tie(0);
ios::sync_with_stdio(0);
int n, k, p;
cin >> n >> k >> p;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
int ans = 0;
for (int i = 1; i <= n; i++)
{
ans *= 2;
ans %= p;
int precnt = (qmi(2, i - 1, p) - 1+p)%p;
int xx = a[i] / k;
xx %= p;
int lf = a[i] - xx*k;
lf %= p;
ans += (precnt*xx)%p;
ans %= p;
ans += xx;
ans %= p;
for (int j = 0; (j + lf) < k; j++)
{
hnx[j + lf] += h[j];
hnx[j + lf] %= p;
}
for (int j = k - 1; (j + lf) >= k; j--)
{
ans += h[j];
ans %= p;
hnx[j + lf-k] += h[j];
hnx[j + lf-k] %= p;
}
hnx[lf]++;
hnx[lf] %= p;
for (int j = 0; j <= k - 1; j++)
{
h[j] = hnx[j];
}
}
ans = (ans%p + p) % p;
cout << ans << '\n';
system("pause");
return 0;
}