[DP][UNR #2] 피로가 쌓여 병이 되다

fi, j f i, j는 길이가 ii 최대치가 jj인 서열의 답안을 나타낸다
최대치를 일일이 들고 자리를 옮기면 돼요.

#include 
#include 
#include 

using namespace std;

const int N=410,P=998244353;

int n,k,ans,a[N],f[N][N],g[N][N],pw[N][N],val[N][N];

inline int Pow(int x,int y){
  int ret=1;
  for(;y;y>>=1,x=1LL*x*x%P) if(y&1) ret=1LL*ret*x%P;
  return ret;
}

int main(){
  scanf("%d%d",&n,&k);
  for(int i=1;i<=n;i++){
    scanf("%d",&a[i]);
    pw[i][0]=1;
    for(int j=1;j<=n;j++) pw[i][j]=1LL*pw[i][j-1]*a[i]%P;
  }
  for(int _n=1;_n<=n;_n++)
    for(int i=1;i<=_n;i++)
      for(int j=1;j+k-1<=_n;j++)
    if(j<=i && j+k-1>=i) val[_n][i]++;
  for(int i=1;i<=n;i++)
    f[0][i]=g[0][i]=1;
  for(int i=1;i
    for(int j=1;j<=n;j++)
      f[i][j]=(Pow(j,i)+P-Pow(j-1,i))%P,g[i][j]=Pow(j,i);
  for(int i=k;i<=n;i++)
    f[i][1]=1LL*f[i-1][1]*a[1]%P,g[i][1]=f[i][1];
  for(int i=k;i<=n;i++)
    for(int j=2;j<=n;j++)
      for(int s=1;s<=i;s++){
    f[i][j]=(f[i][j]+1LL*g[s-1][j-1]*pw[j][val[i][s]]%P*g[i-s][j])%P;
    g[i][j]=(g[i][j-1]+f[i][j])%P;
      }
  int ans=0;
  for(int i=1;i<=n;i++) ans=(ans+f[n][i])%P;
  printf("%d\n",ans);
  return 0;
}