【dp】排队

分享一道比较难想的dp题

排队

问题描述

小sin所在的班有n名同学，正准备排成一列纵队，但他们不想按身高从矮到高排，那样太单调，太没个性。他们希望恰好有k对同学是高的在前，矮的在后，其余都是矮的在前，高的在后。如当n=5，k=3时，假设5人从矮到高分别标为1、2、3、4、5，则（1,5,2,3,4）、（2,3,1,5,4）、（3,1,4,2,5）都是可行的排法。小sin想知道总共有多少种可行排法。

输入

输入文件名为lineup.in。
一行两个整数n和k，意义见问题描述。

输出

输出文件名为lineup.out。
输出一个整数，表示可行排法数。由于结果可能很大，请输出排法数mod 1799999的值。

样例输入

5 3

样例输出

15

数据范围

对于20%的数据，有n≤10，k≤40；
对于60%的数据，有n≤100，k≤500；
对于100%的数据，有n≤100，k≤n*(n-1)/2。

这道题我们先从模拟的角度来看，如果我们现在有5个数，1、2、3、4、5，这时我们要将第六个数加进去，即插入6这个数。
显而易见，一共有6种方法，加在最后，加在4,5间，3,4间，2,3间，1,2间，1前面。
如下图

将6放在最后对逆序对数不会造成任何影响。

放在4,5之间逆序对会加1

以此类推，放在最前面会加5

所以我们可以得到递推式
如果用f[i][j]表示有i个数，j个逆序对有多少种情况，所以
f[i][j]=f[i-1][j]+f[i-1][j-1]+……+f[i-1][j-i+1];
但是这个仍然要超时，我们可以优化一下。
f[i][j-1]=f[i-1][j-1]+f[i-1][j-2]+……+f[i-1][j-i];
所以f[i][j]=f[i][j-1]+f[i-1][j]-f[i-1][j-i];

所以程序很容易就可以写出来，代码如下：

#include<iostream>
#include<cstdio>
#define M 101
#define P 1799999

using namespace std;

long long f[M][M*M/2],n,k;

int main()
{
    f[2][1]=1;f[2][2]=1;
    cin>>n>>k;
    for(int i=3;i<=n;i++)
        for(int j=0;j<=(i*(i-1)/2)+1;j++)
        {
            if(j==1) f[i][j]=1;
            else if(j>(i*(i-1)/4)+1)f[i][j]=f[i][i*(i-1)/2-j+2];
            else if (j-i>=1)f[i][j]=(P+f[i-1][j]+f[i][j-1]-f[i-1][j-i])%P;
            else f[i][j]=(f[i-1][j]+f[i][j-1])%P;
        }
    cout<<f[n][k+1];
}

如果有什么问题，或错误，请在评论区提出，谢谢。