UOJ #311「UNR #2」积劳成疾

需要锻炼$ DP$能力

UOJ #311

---

题意

等概率产生一个长度为$ n$且每个数在[1,n]间随机的数列

定义其价值为所有长度为$ k$的连续子数列的最大值的乘积

给定$ n,k$求所有合法数列的价值和

---

题解

设$ f(x,y)$表示长度为$x$的数列中，最值不超过$ y$的所有数列的价值和

若数列的最值不是$ y$则$ f(x,y)=f(x,y-1)$

否则枚举最左边的最值位置，设为位置$ i$

则$ f(x,y)$可由$f(i-1,y-1)·w(y)^{calc(i)}·f(x-i,y)$转移过来

其中$ calc(i)$表示在长度为$ x$的数列中有多少个长度为$ k$的数列包含第$ i$个位置

时间复杂度$ O(n^3)$

---

代码

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#define p 998244353
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x=;char zf=;char ch=getchar();
    while(ch!='-'&&!isdigit(ch))ch=getchar();
    if(ch=='-')zf=-,ch=getchar();
    while(isdigit(ch))x=x*+ch-'',ch=getchar();return x*zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int k,m,n,x,y,z,cnt,ans;
int w[];
int calc(int x,int y){
    int L=max(,x-m+),R=min(x,y-m+);
    return max(,R-L+);
}
int f[][],mi[][];
int main(){
    n=read();m=read();
    for(rt i=;i<=n;i++)w[i]=read();
    for(rt i=;i<=n;i++)f[][i]=;
    for(rt i=;i<=n;i++){
        mi[i][]=;
        for(rt j=;j<=n;j++)mi[i][j]=1ll*mi[i][j-]*w[i]%p;
    }
    for(rt i=;i<=n;i++)
    for(rt j=;j<=n;j++){
        if(j>)f[i][j]=f[i][j-];
        for(rt k=;k<=i;k++)(f[i][j]+=1ll*f[k-][j-]*f[i-k][j]%p*mi[j][calc(k,i)]%p)%=p;
    }
    cout<<f[n][n];
    return ;
}