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

Solution

令 $f_{i,j}$ 表示前 $i$ 个数产生 $j$ 个逆序对的方案数，每次考虑把 $i + 1$ 加入，有 $i + 1$ 个插入位置分别产生 $0 . . i$ 个新的逆序对。

因此 $f_{n}$ 的生成函数为 $(1+x)(1+x+x2)…(1+x+…xn−1)(1+x)(1+x+x^2)\dots (1+x+\dots x^{n-1})$
即 $fn=(1−x)(1−x2)…(1−xn)(1−x)nf_n=\frac{(1-x)(1-x^2)\dots (1-x^n)}{(1-x)^n}$

$1(1−x)n\frac{1}{(1-x)^n}$ 的幂级数为 $∑i(n+i−1i)\sum_{i}\binom{n+i-1}{i}$ ，于是只需要考虑分子的系数。

我们只需要求出 $x0…xkx^0\dots x^k$ 的系数，因此我们至多会选择 $O(k)O(\sqrt k)$ 个 $x^i$ ，于是考虑 $d p$ ，令 $f_{i,j}$ 表示 $i$ 个数总和为 $j$ ，最大数不超过 $n$ 的方案数，每次可以整体加 $1$ 或者添加一个 $" 1 "$ 之后再加 $1$ ，有：
$f_{i,j}=f_{i,j-i}+f_{i-1,j-1}-f_{i-1,j-n-1}$

这里的 $f_{i-1,j-n-1}$ 是因为这些数可能会超过 $n$ ，因此我们要把超过 $n$ 的去掉，也就是需要去掉存在 $n + 1$ 的不合法方案。

到这里就可以合并两个幂级数求出答案了，注意原式中是 $x^i$ ，因此系数要乘上一个 $1)^i$ 。

时间复杂度 $O(kk)O(k\sqrt k)$

Code

#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <ctime>
#include <cassert>
#include <string.h>
//#include <unordered_set>
//#include <unordered_map>
//#include <bits/stdc++.h>

#define MP(A,B) make_pair(A,B)
#define PB(A) push_back(A)
#define SIZE(A) ((int)A.size())
#define LEN(A) ((int)A.length())
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define fi first
#define se second

using namespace std;

template<typename T>inline bool upmin(T &x,T y) { return y<x?x=y,1:0; }
template<typename T>inline bool upmax(T &x,T y) { return x<y?x=y,1:0; }

typedef long long ll;
typedef unsigned long long ull;
typedef long double lod;
typedef pair<int,int> PR;
typedef vector<int> VI;

const lod eps=1e-11;
const lod pi=acos(-1);
const int oo=1<<30;
const ll loo=1ll<<62;
const int mods=1e9+7;
const int MAXN=200005;
const int INF=0x3f3f3f3f;//1061109567
/*--------------------------------------------------------------------*/
inline int read()
{
	int f=1,x=0; char c=getchar();
	while (c<'0'||c>'9') { if (c=='-') f=-1; c=getchar(); }
	while (c>='0'&&c<='9') { x=(x<<3)+(x<<1)+(c^48); c=getchar(); }
	return x*f;
}
int fac[MAXN],inv[MAXN],g[505][MAXN];
int upd(int x,int y) { return x+y>=mods?x+y-mods:x+y; }
int quick_pow(int x,int y)
{
	int ret=1;
	for (;y;y>>=1)
	{
		if (y&1) ret=1ll*ret*x%mods;
		x=1ll*x*x%mods;
	}
	return ret;
}
int C(int x,int y) { return 1ll*fac[x]*inv[y]%mods*inv[x-y]%mods; }
void Init(int n)
{
	fac[0]=1;
	for (int i=1;i<=n;i++) fac[i]=1ll*fac[i-1]*i%mods;
	inv[n]=quick_pow(fac[n],mods-2);
	for (int i=n-1;i>=0;i--) inv[i]=1ll*inv[i+1]*(i+1)%mods;
}
signed main()
{
	int n=read(),k=read(),sz=sqrt(k*2)+10; Init(n+k);
	g[0][0]=1;
	for (int i=1;i<=sz;i++)
		for (int j=i;j<=k;j++) 
			g[i][j]=upd(upd(g[i][j-i],g[i-1][j-i]),mods-(j>=n+1?g[i-1][j-n-1]:0));
	int ans=0;
	for (int i=0;i<=sz;i++) 
		for (int j=0;j<=k;j++) 
			if (i&1) ans=upd(ans,mods-1ll*C(n+j-1,j)*g[i][k-j]%mods);
			else ans=upd(ans,1ll*C(n+j-1,j)*g[i][k-j]%mods);
	printf("%d\n",ans);
	return 0;
}