제출 #1097958

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1097958		LM1	Sum Zero (RMI20_sumzero)	C++14	100 / 100	167 ms	19068 KiB

sumzero

#include<bits/stdc++.h>
#define ll long long
#define pii pair<int,int>
//#define int long long
#define pb push_back
#define vi vector<int>
#define ff first
#define ss second
using namespace std;
const int N=4e5+5;
int nxt[N],ans[N],p1[N],q[N],q1[N];
int n,Q,i,jump,j;
ll sum;
unordered_map<ll,int>sf;
int main(){
	ios_base::sync_with_stdio(NULL);cin.tie(NULL);
	cin>>n;
	for(i=1;i<=n;i++)cin>>p1[i];
	sf[0]=n+1;
	nxt[n+1]=n+2;
	nxt[n+2]=n+2;
	for(i=n;i>=1;i--){
		sum+=p1[i];
		if(sf.find(sum)==sf.end())nxt[i]=nxt[i+1];
		else nxt[i]=min(sf[sum],nxt[i+1]);
		sf[sum]=i;
	}
	cin>>Q;
	for(int i=0;i<Q;i++)cin>>q[i]>>q1[i];
	for(jump=18;jump>=0;jump--){
		for(i=1;i<=n+2;i++)p1[i]=nxt[i];
		for(j=1;j<=jump;j++){
			for(i=1;i<=n+2;i++)p1[i]=p1[p1[i]];
		}
		for(i=0;i<Q;i++){
			if(p1[q[i]]<=q1[i]+1){
				ans[i]+=(1<<jump);
				q[i]=p1[q[i]];
			}
		}
	}
	for(i=0;i<Q;i++)cout<<ans[i]<<"\n";
}
/*
dp[i]=dp[i-1]
pr[i]
ind=mp[pr[i]]
dp[i]=max(dp[i],dp[ind-1]+1)
mp[pr[i]]=i
 
nxt
i, j
i<=k<=j
pr[k,j]=0
l,r
ans=0;
while(l<=r){
	ans++;
	l=nxt[l];
}
 
p[i][0]=nxt[i]
for(int j=1;j<19;j++){
	for(int i=0;i<n;i++){
//		i ze xar 
//		2^j cali next
		p[i][j]=p[p[i][j-1][j-1]
	}
}
l,r
for(int j=18;j>=0;j--){
	if(p[l][j]<=r){
		l=p[l][j]
		ans+=(1<<j)
	}
}
p[i][j]=p[p[i][j-1][j-1];
x=p[i][j-1]
p[x][j-1]
nxt[i]
*/

• Subtask 1: 22 / 22
• Subtask 2: 39 / 39
• Subtask 3: 39 / 39