The 2018 ACM-ICPC China JiangSu Provincial Programming Contest - B. Array (DP+滚动数组+离线)

https://nanti.jisuanke.com/t/28866

 

题意：

每次给你n、k，问你1-n的排列恰好有k个逆序对的排列有多少种。

n、k<=5000。

 

POINT：

可知：dp[n][k]的递推公式为

比如dp[4][4]=dp[3][1]+dp[3][2]+dp[3][3]+dp[3][4]。 

为什么dp[4][4]不能从dp[3][0]得到，因为把4放在【1-3的排列】的最前面，最多只能得到3个贡献。

所以每次dp[i][j]都是从前一个的j-i+1 到j。 当然，j-i+1要和0做比较。比如dp[4][2]=dp[3][0]+dp[3][1]+dp[3][2]。别越界了。

所以每次加的是一个区间，可以每次加一个，减一个得到。

之后5000×5000的数组开不下，由递推式可知dp只和上一个有关，所有可以用滚动数组。

因而离线。

 

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
const int maxn = 5050;
int dp[10][5010];
const int mod = 1e9+7;
struct node
{
    int n,k;
    int id;
    bool friend operator < (node a,node b)
    {
        return a.n<b.n;
    }
}query[maxn];

int ans[maxn];
int main()
{
    int cnt=0,n,k;
    while(~scanf("%d%d",&n,&k)){
        query[++cnt].n=n;
        query[cnt].k=k;
        query[cnt].id=cnt;
        if(n==0) break;
    }
    sort(query+1,query+1+cnt);
    dp[0][0]=1;
    int p=1;
    int now=1;
    for(int i=1;i<=5000;i++){
        int sum=0;
        for(int j=0;j<=5000;j++){
            sum=(sum+dp[p^1][j])%mod;
            if(j-i>=0)
                sum=(sum-dp[p^1][j-i]+mod)%mod;
            dp[p][j]=sum;
        }
        while(query[now].n==i){
            ans[query[now].id]=dp[p][query[now].k];
            now++;
        }
        p=p^1;
    }
    for(int i=1;i<=cnt;i++){
        printf("%d\n",ans[i]);
    }



    return 0;
}