[2018雅礼集训1-12]小C的线段树 DP

题面
考虑当$n>m$时答案为$0$，$1\le nm \le 10^5$，所以只用考虑$n<10^{\frac{5}{2}}$的情况。
把一个区间看成一对括号，设$f_{i,l,r}$表示前$i$个点，之前有$l$个左括号，$r$个右括号，$g_{i,l,r}$表示其方案数，那么一个点的贡献就是$(l-r)^k$，然后分这个点不放、放左括号、放右括号、放两个括号转移即可。复杂度$O(n^2m)$。
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
const int mod=998244353;
int n,m,k;
ll ans,f[2][333][333],g[2][333][333],mi[100010];
ll ksm(ll a,int b){ll r=1;for(;b;b>>=1){if(b&1)r=r*a%mod;a=a*a%mod;}return r;}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    if(n>m) {puts("0");return 0;}
    for(int i=0;i<=m;i++) mi[i]=ksm(i,k);
    g[0][0][0]=1;
    for(int i=1,v=1;i<=m;i++,v^=1)
        for(int l=0;l<=min(n,i);l++)
            for(int r=0;r<=l;r++)
            {
                g[v][l][r]=f[v][l][r]=0;
                g[v][l][r]+=g[v^1][l][r];f[v][l][r]+=f[v^1][l][r];
                if(l)g[v][l][r]+=g[v^1][l-1][r],f[v][l][r]+=f[v^1][l-1][r];
                if(r)g[v][l][r]+=g[v^1][l][r-1],f[v][l][r]+=f[v^1][l][r-1];
                if(l&&r)g[v][l][r]+=g[v^1][l-1][r-1],f[v][l][r]+=f[v^1][l-1][r-1];
                (f[v][l][r]+=mi[l-r]*(g[v][l][r]%=mod))%=mod;
            }   
    printf("%lld",f[m&1][n][n]);
    return 0;
}