不難發現，一個集合S的權值即等於 ∏x∈S(x+1)
所以題目轉化成 1,2,3,...,n 個數中取若干個數加和為 P 的方案數。
這個可以 O(nn−−√) 的 DP，比較經典：
f[i][j] 表示取了i個數，加和為j的方案，轉移時考慮把所有數加1， 或新取一個數1即可。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100005,MOD=998244353;
int n,m,f[450
 
][maxn],ans;
int main(){
    freopen("B.in","r",stdin);
    freopen("B.out","w",stdout);
    scanf("%d%d",&n,&m); n=min(n,m); int sqrtm=sqrt(m*2)+1;
    f[0][0]=1;
    for(int i=1;i<=sqrtm;i++)
     for(int j=i;j<=m;j++){
        (f[i][j]+=f[i][j-i]+f[i-1][j-i])%=MOD;
        if(j>=n+1
 
) (f[i][j]-=f[i-1][j-(n+1)])%=MOD;
     }
    for(int i=1;i<=sqrtm;i++) ans=(ans+f[i][m])%MOD;
    printf("%d\n",(ans+MOD)%MOD);
}