loj #6077. 「2017 山东一轮集训 Day7」逆序对

 

题目描述

给定 n,k n, kn,k，请求出长度为 n nn 的逆序对数恰好为 k kk 的排列的个数。答案对 109+7 10 ^ 9 + 710​9​​+7 取模。

对于一个长度为 n nn 的排列 p pp，其逆序对数即满足 i<j i < ji<j 且 pi>pj p_i > p_jp​i​​>p​j​​ 的二元组 (i,j) (i, j)(i,j) 的数量。

输入格式

一行两个整数 n,k n, kn,k。

输出格式

一行，表示答案。

样例

样例输入

7 12

样例输出

531

数据范围与提示

对于 20% 20\%20% 的数据，n,k≤20 n, k \leq 20n,k≤20；
对于 40% 40\%40% 的数据，n,k≤100 n, k \leq 100n,k≤100；
对于 60% 60\%60% 的数据，n,k≤5000 n, k \leq 5000n,k≤5000；
对于 100% 100\%100% 的数据，1≤n,k≤100000,1≤k≤(n2) 1 \leq n, k \leq 100000, 1 \leq k \leq \binom{n}{2}1≤n,k≤100000,1≤k≤(​2​n​​)。

 

 

#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 1000000007
#define maxn 5002
using namespace std;
int dp[maxn][maxn],n,sum[maxn][maxn],m;
int Max(int x,int y){return x>y?x:y;}
int Min(int x,int y){return x<y?x:y;}
int main(){
    scanf("%d%d",&n,&m);m++;
    dp[1][1]=1;
    sum[1][1]=1;
    int lim,i,j,l,r;
    for(i=2;i<=n;++i){
        lim=(i-1)*i/2+1;
        for(j=1;j<=min(lim,m);++j){
            l=Max(j-i,0);r=Min(j,(i-1)*(i-2)/2+1);
            dp[i][j]=sum[i-1][r]-sum[i-1][l];
            if(dp[i][j]<0)dp[i][j]+=mod;
            else if(dp[i][j]>=mod)dp[i][j]-=mod;
            sum[i][j]=sum[i][j-1]+dp[i][j];
            if(sum[i][j]>=mod)sum[i][j]-=mod;
        }
    }
    printf("%d",dp[n][m]);
    return 0;
}

60分 dp+前缀和优化