QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#143182#6509. Not Another Range Query ProblemqzezWA 71ms360360kbC++145.4kb2023-08-20 20:48:452023-08-20 20:48:47

Judging History

你现在查看的是最新测评结果

  • [2023-08-20 20:48:47]
  • 评测
  • 测评结果:WA
  • 用时:71ms
  • 内存:360360kb
  • [2023-08-20 20:48:45]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
template<typename T>
ostream& operator << (ostream &out,const vector<T>&x){
	if(x.empty())return out<<"[]";
	out<<'['<<x[0];
	for(int len=x.size(),i=1;i<len;i++)out<<','<<x[i];
	return out<<']';
}
template<typename T>
ostream& operator << (ostream &out,const deque<T>&x){
	if(x.empty())return out<<"[]";
	out<<'['<<x[0];
	for(int len=x.size(),i=1;i<len;i++)out<<','<<x[i];
	return out<<']';
}
template<typename T>
vector<T> ary(const T *a,int l,int r){
	return vector<T>{a+l,a+1+r};
}
template<typename T>
void debug(T x){
	cerr<<x<<'\n';
}
template<typename T,typename ...S>
void debug(T x,S ...y){
	cerr<<x<<' ',debug(y...);
}
const int N=5e5+10,INF=1e9;
int n,m,nex[N],pre[N];
char a[N];
mt19937 rnd(0);
struct tree{
	int rnd,ls,rs,siz,mn,sum;
	deque<int>A;
}t[N];
int root;
void pushup(int rt){
	t[rt].siz=t[t[rt].ls].siz+t[t[rt].rs].siz+1;
	t[rt].mn=min({t[t[rt].ls].mn,t[t[rt].rs].mn,(int)t[rt].A.size()});
	t[rt].sum=t[t[rt].ls].sum+t[t[rt].rs].sum+(int)t[rt].A.size();
}
void split_r(int rt,int R,int &x,int &y){
	if(!rt){
		x=y=0;return;
	}
	if(t[rt].A.back()<=R)x=rt,split_r(t[rt].rs,R,t[rt].rs,y);
	else y=rt,split_r(t[rt].ls,R,x,t[rt].ls);
	pushup(rt);
}
// void split_l(int rt,int L,int &x,int &y){
// 	if(!rt){
// 		x=y=0;return;
// 	}
// 	if(t[rt].A.front()<=L)x=rt,split_l(t[rt].rs,L,t[rt].rs,y);
// 	else y=rt,split_l(t[rt].ls,L,x,t[rt].ls);
// 	pushup(rt);
// }
int merge(int x,int y){
	if(!x||!y)return x|y;
	if(t[x].rnd<t[y].rnd){
		t[x].rs=merge(t[x].rs,y);
		return pushup(x),x;
	}else{
		t[y].ls=merge(x,t[y].ls);
		return pushup(y),y;
	}
}
int cnt,cur[N];
void rebuild(int rt){
	if(!rt)return;
	rebuild(t[rt].ls);
	cur[++cnt]=rt;
	rebuild(t[rt].rs);
	t[rt].ls=t[rt].rs=0;
}
struct ques{
	int l,r,k,id;
}o[N];
int ans[N];
void init(){
	t[0].mn=INF;
	int k=0;
	root=0;
	for(int l=1,r;l<=n;l=r){
		for(r=l+1;r<=n&&a[l]==a[r];r++);
		for(int i=l;i+1<r;i++)nex[i]=i+1,pre[i+1]=i;
		pre[l]=nex[r-1]=0;
		t[++k]={(int)rnd(),0,0,1,r-l,r-l,{}};
		// debug(t[k].A);
		for(int i=l;i<r;i++)t[k].A.push_back(i);
		root=merge(root,k);
	}
}
void print(int rt=root){
	if(!rt)return;
	print(t[rt].ls);
	vector<int>A;
	cerr<<t[rt].A<<' ';
	print(t[rt].rs);
}
void solve1(){
	init();
	// debug("OK1");
	// debug(ary(nex,1,n));
	// debug(ary(pre,1,n));
	for(int now=0,x=1;root;){
		// print(),cerr<<endl;
		int mn=t[root].mn;
		// debug(now,mn);
		for(;x<=m&&o[x].k<now+mn;x++){
			int r1,r2,r3;
			split_r(root,o[x].r-1,r1,r2);
			for(r3=r2;t[r3].ls;r3=t[r3].ls);
			// debug("ques",t[r1].sum,t[r1].siz,o[x].k,now);
			int ans=t[r1].sum-t[r1].siz*(o[x].k-now);
			// debug("ans",ans,o[x].r);
			if(r3){
				// debug("A",t[r3].A);
				int cnt=upper_bound(t[r3].A.begin(),t[r3].A.end(),o[x].r)-t[r3].A.begin();
				ans+=max(cnt-(o[x].k-now),0);
			}
			// debug("ans",ans);
			::ans[o[x].id]+=ans;
			root=merge(r1,r2);
		}
		cnt=0,rebuild(root),root=0;
		// debug("cur",cnt,ary(cur,1,cnt));
		int tot=0;
		for(int i=1;i<=cnt;i++){
			int rt=cur[i];
			if(t[rt].A.size()==mn)continue;
			for(int j=1;j<=mn;j++)t[rt].A.pop_front();
			if(!tot||a[t[cur[tot]].A.back()]!=a[t[rt].A.front()]){
				cur[++tot]=rt;
			}else{
				int las=cur[tot];
				if(t[las].A.size()<t[rt].A.size()){
					for(int i=t[las].A.size()-1;~i;i--){
						t[rt].A.push_front(t[las].A[i]);
					}
					cur[tot]=rt;
				}else{
					for(int x:t[rt].A){
						t[las].A.push_back(x);
					}
				}
			}
		}
		for(int i=1;i<=tot;i++){
			int rt=cur[i];
			pushup(rt);
			root=merge(root,rt);
		}
		// if(now==1)return;
		now+=mn;
		// print(),cerr<<endl;
		// return;
	}
	// debug("ans1",ans[1]);
}
void solve2(){
	init();
	// debug("OK1");
	// debug(ary(nex,1,n));
	// debug(ary(pre,1,n));
	for(int now=0,x=1;root;){
		// print(),cerr<<endl;
		int mn=t[root].mn;
		// debug(now,mn);
		for(;x<=m&&o[x].k<now+mn;x++)if(o[x].l>1){
			int r1,r2,r3;
			split_r(root,o[x].l-2,r1,r2);
			for(r3=r2;t[r3].ls;r3=t[r3].ls);
			// debug("ques",t[r1].sum,t[r1].siz,o[x].k,now);
			int ans=t[r1].sum-t[r1].siz*(o[x].k-now);
			// debug("ans",ans,o[x].r);
			if(r3){
				// debug("A",t[r3].A);
				int cnt=upper_bound(t[r3].A.begin(),t[r3].A.end(),o[x].l-1)-t[r3].A.begin();
				ans+=max(cnt-(o[x].k-now),0);
			}
			// debug("ans",ans);
			::ans[o[x].id]-=ans;
			root=merge(r1,r2);
		}
		cnt=0,rebuild(root),root=0;
		// debug("cur",cnt,ary(cur,1,cnt));
		int tot=0;
		for(int i=1;i<=cnt;i++){
			int rt=cur[i];
			if(t[rt].A.size()==mn)continue;
			for(int j=1;j<=mn;j++)t[rt].A.pop_back();
			if(!tot||a[t[cur[tot]].A.back()]!=a[t[rt].A.front()]){
				cur[++tot]=rt;
			}else{
				int las=cur[tot];
				if(t[las].A.size()<t[rt].A.size()){
					for(int i=t[las].A.size()-1;~i;i--){
						t[rt].A.push_front(t[las].A[i]);
					}
					cur[tot]=rt;
				}else{
					for(int x:t[rt].A){
						t[las].A.push_back(x);
					}
				}
			}
		}
		for(int i=1;i<=tot;i++){
			int rt=cur[i];
			pushup(rt);
			root=merge(root,rt);
		}
		// if(now==1)return;
		now+=mn;
		// print(),cerr<<endl;
		// return;
	}
}
int main(){
	scanf("%d%d%s",&n,&m,a+1);
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&o[i].l,&o[i].r,&o[i].k);
		o[i].id=i;
	}
	sort(o+1,o+1+m,[](ques x,ques y){return x.k<y.k;});
	solve1();
	solve2();
	for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 65ms
memory: 360028kb

input:

9 7
100110001
2 5 1
3 6 1
4 8 2
2 7 1
1 9 1
1 9 0
1 9 8

output:

2
1
1
3
4
9
0

result:

ok 7 numbers

Test #2:

score: -100
Wrong Answer
time: 71ms
memory: 360360kb

input:

100 100000
0000011010101000111011110110000111110101101010111111101011011010111001111010111000001000011000001010
76 99 3
25 84 7
45 83 11
10 12 10
69 86 4
27 28 1
22 42 42
4 86 25
26 91 22
20 81 17
50 78 0
77 93 50
31 50 34
7 46 13
78 89 0
79 98 0
2 84 33
58 93 11
56 75 2
55 77 68
7 9 41
44 46 11
47 ...

output:

9
14
4
0
3
0
0
0
1
6
29
0
0
0
12
20
0
0
5
0
0
-8
0
-3
-2
0
0
11
18
1
0
57
0
-2
11
0
3
0
0
3
0
0
0
0
0
-1
0
0
-2
0
-5
0
-5
0
19
0
0
0
12
5
0
0
3
-6
3
0
1
10
13
0
0
0
6
0
8
0
1
16
-1
19
29
40
21
12
26
0
21
6
1
10
18
2
4
-1
2
6
0
0
6
0
0
-3
51
0
0
-1
19
11
0
21
6
9
10
0
16
22
0
21
0
26
0
0
0
0
0
0
11
4...

result:

wrong answer 1st numbers differ - expected: '8', found: '9'