首页 > 其他好文 > 详细

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

时间：2017-12-20 15:07:25 阅读：130 评论：0 收藏：0 [点我收藏+]

标签：最大的 submit lin log main printf http sub stat

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]);        
}