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

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

Solution

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

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

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

我们只需要求出x0…xkx^0\dots x^kx0xk的系数,因此我们至多会选择O(k)O(\sqrt k)O(k)−xi-x^ixi,于是考虑dpdpdp,令fi,jf_{i,j}fi,j表示iii个数总和为jjj,最大数不超过nnn的方案数,每次可以整体加111或者添加一个"1""1""1"之后再加111,有:
fi,j=fi,j−i+fi−1,j−1−fi−1,j−n−1f_{i,j}=f_{i,j-i}+f_{i-1,j-1}-f_{i-1,j-n-1}fi,j=fi,ji+fi1,j1fi1,jn1

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

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

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

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值