牛客挑戰賽49 D筱瑪愛線段樹 線段樹+樹狀數組計數

 存下操作, 從後往前 用樹狀數組維護操作的次數, 編號從大到小,防止重複調用遞歸。

然後再遍歷一次,操作一的次數對線段樹修改,大概 m*O(logn)

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 100005;
const int mod = 1e9+7;
ll b[maxn];
struct N{
	int l,r;
	ll pre,add;
}node[4*maxn+5];
struct Q{
	int op,l,r;
}q[maxn];
int lowbit(int k)
{
	return k&(-k);
}
void update(int k,ll val)
{
	while(k<maxn){
		b[k]+=val;
		k+=lowbit(k);
	}
}
ll sum(int k)
{
   ll ans = 0;
   while(k>0){
   		ans=(ans+(ll)b[k])%mod; 
   		 k-=lowbit(k);
   }
   return ans%mod;
	
}
void build(int p,int l,int r){//區間編號p 
	node[p].l = l;
        node[p].r = r;
        node[p].add = 0;
	if(l==r){//葉子結點直接賦值 
		node[p].pre = 0;
		return ;
	}
	int mid = l+r >>1;
	build(2*p,l,mid);
	build(2*p+1,mid+1,r);
	node[p].pre = node[p*2].pre + node[p*2+1].pre;//維護區間和 
}
void spread(int p){
	if(node[p].add){//如果標記不爲0 向下傳遞 
		node[2*p].pre+=(node[2*p].r - node[2*p].l + 1)*node[p].add%mod;
		node[2*p].pre%=mod;		
		node[2*p+1].pre+=(node[2*p+1].r - node[2*p+1].l + 1)*node[p].add%mod;
		node[2*p+1].pre%=mod;
		node[2*p].add+= node[p].add;
		node[2*p+1].add+= node[p].add;
		node[p].add = 0;
	}
}
void change(int p,int l,int r,int v){
	if(node[p].l>=l&&node[p].r<=r){//區間被覆蓋 就對其進行修改 
		node[p].pre+=((ll)v*(node[p].r-node[p].l+1))%mod;
		node[p].pre%=mod;
		node[p].add+= v; //加上懶惰標記 
		return ;
	}
	spread(p);//沒有覆蓋 下放懶惰標記 考慮兒子區間可能因爲懶惰標記存在而沒有修改
	int mid = node[p].l + node[p].r>>1; 
	if(l<=mid)change(p*2,l,r,v);
	if(r>mid)change(p*2+1,l,r,v);
	node[p].pre = (node[2*p].pre + node[2*p+1].pre)%mod;
} 
ll query(int p,int l,int r){
	if(l<=node[p].l&&r>=node[p].r){//被覆蓋 返回值 
		return node[p].pre;
	}
	spread(p);
	ll ans = 0;
	int mid = node[p].r+node[p].l>>1;
	 if(l<=mid)ans+=query(p*2,l,r),ans%=mod;
	 if(r>mid)ans+=query(p*2+1,l,r),ans%=mod;
	 return ans%mod;
}
int main()
{
	int n,m;
	scanf("%d %d",&n,&m);	
	build(1,1,n);
	for(int i=1;i<=m;i++){
		scanf("%d %d %d",&q[i].op,&q[i].l,&q[i].r);
	}

	for(int i=m;i>=1;i--){
		if(q[i].op==2){//維護  第 i 條指令 對指令 l r 操作 的次數  
			int val = (sum(i)+1)%mod;
			update(q[i].l,val);
			update(q[i].r+1,-val);
		}
	}
	for(int i=1;i<=m;i++){
		if(q[i].op==1){
			int val = (sum(i)+1)%mod;
			change(1,q[i].l,q[i].r,val);		 
		}
	}
	for(int i = 1;i<=n;i++){
		printf("%lld%c",query(1,i,i)%mod," \n"[i==n]);
	}
	return 0;
 } 
 /*
 5 5
 1 1 1
 2 1 1
 2 1 2
 2 1 3
 2 1 4
 
 */

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章