bzoj千题计划153：bzoj2431: [HAOI2009]逆序对数列

Posted 2020-10-18 日拱一卒功不唐捐

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了bzoj千题计划153：bzoj2431: [HAOI2009]逆序对数列相关的知识，希望对你有一定的参考价值。

http://www.lydsy.com/JudgeOnline/problem.php?id=2431

dp[i][j] 表示i的排列，有j个逆序对的方案数

加入i+1，此时i+1是排列中最大的数，

所以放在i+1后面的所有数都会与i+1形成逆序对

转移方程：dp[i][j]=Σ dp[i-1][j-k] k∈[0,min(j,i-1)]

前缀和优化

朴素的DP

#include<cstdio>
#include<algorithm>

using namespace std;

const int mod=10000;

int dp[1001][1001];

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    dp[1][0]=1;
    int m;
    for(int i=2;i<=n;++i)
    {
        dp[i][0]=1;
        m=min(i*(i-1)/2,k);
        for(int j=1;j<=m;++j)
            for(int k=0;k<=i-1 && k<=j;++k)
                dp[i][j]=(dp[i][j]+dp[i-1][j-k])%mod;
    }
    printf("%d",dp[n][k]);
}

前缀和优化：

#include<cstdio>
#include<algorithm>

using namespace std;

const int mod=10000;

#define N 1001

int dp[N][N],sum[N][N];

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    dp[1][0]=1;
    for(int i=0;i<=k;++i) sum[1][i]=1;
    int m;
    for(int i=2;i<=n;++i)
    {
        dp[i][0]=1;
        sum[i][0]=1;
        m=min(i*(i-1)/2,k);
        for(int j=1;j<=m;++j) 
        {
            if(j<min(j,i-1)+1) dp[i][j]=sum[i-1][j];
            else dp[i][j]=sum[i-1][j]-sum[i-1][j-min(j,i-1)-1];
            if(dp[i][j]<0) dp[i][j]+=mod;
            sum[i][j]=sum[i][j-1]+dp[i][j];
            if(sum[i][j]>0) sum[i][j]-=mod;
        }
        for(int j=i*(i-1)/2+1;j<=k;++j) sum[i][j]=sum[i][j-1];
    }
    printf("%d",dp[n][k]);        
}