题目描述:
小K迷上了一款游戏,对某个敌人进行了连续N天的持续讨伐。
大部分时候,小K可以选择在白天的攻坚之后回到营地或是留在前线,但其中M天由于战况或是气候等特殊原因,小K只能选择留在前线。
不同的休整方案对小K产生的疲劳程度当然是不一样的,具体定义为每一段留在前线持续攻坚的天数的K次方的乘积。小K想知道所有休整方案的疲劳程度之和,对998244353取模,两种休整方案被认为是不同的,当存在一天两种方案选择的休整地点不同。
对于40%的数据:N<=50,M<=10,0<=K<=2
对于100%的数据:N<=10^9,M<=100000,0<=K<=8
样例解释:
n=3,总共只有2个晚上,因为第二晚不能回去,因此只有第一晚可以选择是否回去,如果不回去,总的疲劳度是3^2,如果回去,疲劳度是2^2,求和等于13。
收起
输入
第一行包含三个正整数N,M,K。 第二行包含M个正整数T_1,T_2,...,T_M,表示因为特殊原因无法回营地的日期,保证1<=T_1<T_2<...<T_M<N。
输出
共一行包含一个非负整数Ans,表示所有方案的疲劳程度之和,对998244353取模。
输入样例
3 1 2 2
输出样例
13
题解:
挣扎着卡了卡常过了
记dp[i][j]dp[i][j]为到第ii位,和第ii位相连的部分长度xjxj乘上之前部分所有方案xKxK总和
转移用二项式定理展开即可,若这位不选,可以有dp[i+1][j]=dp[i][K]dp[i+1][j]=dp[i][K]
矩阵乘法优化一下,卡常用预处理出2的几次幂的矩阵的答案。
#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define eps 1e-10
#define ba 47
#define MAXN 2005
//#define ivorysi
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
template<class T>
void read(T &res) {
res = 0;T f = 1;char c = getchar();
while(c < '0' || c > '9') {
if(c == '-') f = -1;
c = getchar();
}
while(c >= '0' && c <= '9') {
res = res * 10 +c - '0';
c = getchar();
}
res *= f;
}
template<class T>
void out(T x) {
if(x < 0) {x = -x;putchar('-');}
if(x >= 10) {
out(x / 10);
}
putchar('0' + x % 10);
}
const int MOD = 998244353;
int N,M,K;
int C[15][15];
bool vis[55];
int inc(int a,int b) {
return a + b >= MOD ? a + b - MOD : a + b;
}
int mul(int a,int b) {
return 1LL * a * b % MOD;
}
void update(int &x,int y) {
x = inc(x,y);
}
struct Matrix {
int f[9][9];
Matrix() {memset(f,0,sizeof(f));}
friend Matrix operator * (const Matrix &a,const Matrix &b) {
Matrix c;
for(int i = 0 ; i <= K ; ++i) {
for(int j = 0 ; j <= K ; ++j) {
for(int h = 0 ; h <= K ; ++h) {
update(c.f[i][j],mul(a.f[i][h],b.f[h][j]));
}
}
}
return c;
}
void unit() {
for(int i = 0 ; i <= K ; ++i) f[i][i] = 1;
}
friend Matrix fpow(Matrix a,int c) {
Matrix res,t = a;res.unit();
while(c) {
if(c & 1) res = res * t;
t = t * t;
c >>= 1;
}
return res;
}
}A,B,ans,P[35];
void Solve() {
read(N);read(M);read(K);
for(int i = 0 ; i <= K ; ++i) {
C[i][0] = 1;
for(int j = 1 ; j <= i ; ++j) {
C[i][j] = inc(C[i - 1][j - 1],C[i - 1][j]);
}
}
for(int i = 0 ; i <= K ; ++i) {
for(int j = 0 ; j <= i ; ++j) {
update(A.f[j][i],C[i][j]);
}
}
B = A;
for(int i = 0 ; i <= K ; ++i) update(B.f[K][i],1);
P[0] = B;
for(int i = 1 ; i <= 30 ; ++i) P[i] = P[i - 1] * P[i - 1];
int p = 1,t;
ans.unit();
for(int i = 1 ; i <= M ; ++i) {
read(t);
for(int j = 0 ; j <= 29 ; ++j) {
if((t - p) >> j & 1) ans = ans * P[j];
}
p = t;
ans = ans * A;++p;
}
if(p != N) {
for(int j = 0 ; j <= 29 ; ++j) {
if((N - p) >> j & 1) ans = ans * P[j];
}
}
int res = 0;
for(int i = 0 ; i <= K ; ++i) {
update(res,ans.f[i][K]);
}
out(res);enter;
}
int main(){
#ifdef ivorysi
freopen("f1.in","r",stdin);
#endif
Solve();
return 0;
}