Newcoder 147 H.Prefix Sum（组合数学+BIT）

最新推荐文章于 2020-08-19 13:20:41 发布

v5zsq

最新推荐文章于 2020-08-19 13:20:41 发布

阅读量233

点赞数

分类专栏： Newcoder 组合数学树状数组

本文链接：https://blog.csdn.net/V5ZSQ/article/details/82844451

版权

组合数学同时被 3 个专栏收录

190 篇文章 1 订阅

订阅专栏

Newcoder

186 篇文章 0 订阅

订阅专栏

树状数组

54 篇文章 0 订阅

订阅专栏

Description

对于一个 $(k+1)\cdot n$ 的矩阵，行编号从 $0$ ~ $k$ ，列编号从 $1$ 到 $n$ ，每次修改第一行元素，之后通过递推式
$a_{i,1}=a_{i-1,1},a_{i,j}=a_{i,j-1}+a_{i-1,j},j\ge 2$
得到整个矩阵 $a_{i,j}$ ，初始状态 $a_{0,1}=...=a_{0,n}=0$ ，给出 $m$ 次操作，操作分两种：

$0\ x\ y:$ 给 $a_{0,x}$ 加上 $y$

$1\ x:$ 查询 $a_{k,x}$

Input

第一行输入三个整数 $n, m, k$ ，之后 $m$ 行每行一个操作

$(1\le n,m\le 10^5,1\le k\le 40,1\le x\le n,1\le y<10^9+7)$

Output

对于每次查询操作，输出 $a_{k,x}$ ，结果模 $10^9+7$

Sample Input

4 11 3
0 1 1
0 3 1
1 1
1 2
1 3
1 4
0 3 1
1 1
1 2
1 3
1 4

Sample Output

1
3
7
13
1
3
8
16

Solution

简单分析可知 $a[k][x]=\sum\limits_{i=1}^xC_{x-i+k-1}^{k-1}\cdot a[0][i]$ ，考虑恒等式 $C_{n+m}^x=\sum\limits_{i=0}^xC_n^iC_m^{x-i}$ ，进而有
$a[k][x]=\sum\limits_{i=1}^x\sum\limits_{j=0}^{k-1}C_x^jC_{k-1-i}^{k-1-j}a[0][x]=\sum\limits_{j=0}^{k-1}C_x^j\sum\limits_{i=1}^xC_{k-1-i}^{k-1-j}a[0][i]$
显然第二个求和只是一个对 $i$ 的前缀求和，故可以用 $k$ 个树状数组维护 $C_{k-1-i}^{k-1-j}a[0][i]$ 的前缀和，单次查询复杂度 $O(klog_n)$ ，总时间复杂度 $O(kmlog_n)$

Code

#include<cstdio>
using namespace std;
typedef long long ll;
#define maxn 100005
#define mod 1000000007
int mul(int x,int y)
{
	ll z=1ll*x*y;
	return z-z/mod*mod;
}
int add(int x,int y)
{
	x+=y;
	if(x>=mod)x-=mod;
	return x;
}
int inv[maxn];
void init(int n=1e5)
{
	inv[1]=1;
	for(int i=2;i<=n;i++)inv[i]=mul(mod-mod/i,inv[mod%i]);
}
struct BIT 
{
	#define lowbit(x) (x&(-x))
	int b[maxn],n;
	void init(int _n)
	{
		n=_n;
		for(int i=1;i<=n;i++)b[i]=0;
	}
	void update(int x,int v)
	{
		while(x<=n)
		{
			b[x]=add(b[x],v);
			x+=lowbit(x);
		}
	}
	int query(int x)
	{
		int ans=0;
		while(x)
		{
			ans=add(ans,b[x]);
			x-=lowbit(x);
		}
		return ans;
	}
}bit[40];
int C(int n,int m)
{
	n=add(n,mod);
	int ans=1;
	for(int i=1;i<=m;i++)
	{
		int temp=add(n,mod-(m-i));
		ans=mul(ans,mul(temp,inv[i]));
	}
	return ans;
}
int main()
{
	init();
	int n,m,k;
	scanf("%d%d%d",&n,&m,&k);
	for(int i=0;i<k;i++)bit[i].init(n);
	while(m--)
	{
		int op,x,y;
		scanf("%d%d",&op,&x);
		if(!op)
		{
			scanf("%d",&y);
			for(int i=0;i<k;i++)bit[i].update(x,mul(C(k-x-1,k-i-1),y));
		}
		else
		{
			int ans=0;
			for(int i=0;i<k;i++)ans=add(ans,mul(C(x,i),bit[i].query(x)));
			printf("%d\n",ans);
		}
	}
	return 0;
}