Submission #978967

#	Time	Username	Problem	Language	Result	Execution time	Memory
978967		NintsiChkhaidze	Sum Zero (RMI20_sumzero)	C++17	0 / 100	12 ms	21336 KiB

sumzero

#include <bits/stdc++.h>
#define ll long long
#define s second
#define f first
#define pb push_back
#define pii pair <int,int>
using namespace std;
 
const int N = 4e5 + 5;
 
int p[N],last[N];
vector <int> v[N],path;
vector <pii> qr[N];

void dfs(int x){
	p[x] = -1;	
	path.pb(x);

	for (auto [y,i]: qr[x]){
		int l = 0,r = (int)path.size() - 1,res=-1;
		while (l <= r){
			int mid = (l + r)/2;
			if (path[mid] <= y){
				res = mid;
				r = mid - 1;
			}else{
				l = mid + 1;
			}
		}

		last[i] = (int)path.size() - res - 1;
	}

	for (int to: v[x])
		dfs(to);

	path.pop_back();
}
signed main(){
	ios_base::sync_with_stdio(0),cin.tie(NULL),cout.tie(NULL);
 
	int n;
	cin>>n;
 
	vector <int> vec;
	map <int,int> mp;
	vec.pb(0);
	for(int i=1;i<=n;i++){
		cin>>p[i];
	}
 
	for (int i = n; i >= 1; i--){
		p[i]+=p[i+1];
		vec.pb(p[i]);
	}
 
	sort(vec.begin(),vec.end());
	int val=0;
	for (int i=0;i<vec.size();i++){
		if (!i || vec[i] != vec[i - 1]) {
			++val;
			mp[vec[i]] = val;
		}
	}
 
	for (int i = 1; i <= n+1; i++)
		p[i] = mp[p[i]],last[p[i]] = 1e9;
 
	last[mp[0]] = n + 1;
	int k = 1e9;
	
	for (int i = n; i >= 1; i--){
		k = min(k,last[p[i]]);
		if (k != 1e9) v[k].pb(i);
		last[p[i]] = i;
	}

	int q;
	cin>>q;
	for (int i = 1; i <= q; i++){
		int l,r;
		cin>>l>>r;
		qr[l].pb({r + 1,i});
	}

	for (int i = n+1; i >= 1; i--){
		if (p[i] != -1) dfs(i);
	}

	for (int i = 1; i <= q; i++)
		cout<<last[i]<<endl;
}

Compilation message (stderr)

sumzero.cpp: In function 'int main()':
sumzero.cpp:59:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |  for (int i=0;i<vec.size();i++){
      |               ~^~~~~~~~~~~

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