Help in optimizing this cs academy problem .

#	User	Rating
1	tourist	3985
2	jiangly	3814
3	jqdai0815	3682
4	Benq	3529
5	orzdevinwang	3526
6	ksun48	3517
7	Radewoosh	3410
8	hos.lyric	3399
9	ecnerwala	3392
9	Um_nik	3392

#	User	Contrib.
1	cry	169
2	maomao90	162
2	Um_nik	162
4	atcoder_official	160
5	djm03178	158
5	-is-this-fft-	158
7	adamant	155
8	Dominater069	154
9	awoo	152
10	luogu_official	150

The problem asks to toggle bits in a given range that after certain operations all bits becomes 0 .

My O(N2) solution is for a given light bulb if it is on the switch it off and update all the given ranges , but this will be N^2 complexity . How can we reduce it to O(n) . The last lines of the editorial i couldn't understood , how the author is updating the ranges .

how to do it . ?

Can anyone help me what the author is done in this code

#include <bits/stdc++.h>
#include <random>
using namespace std;

#define rep(i, N) for (int i = 0; i < N; i++)
#define pb push_back

typedef long long ll;
typedef pair<int, int> i_i;
typedef pair<ll, int> ll_i;
typedef vector<int> vi;

const int MOD = 1e9 + 7;
const int INF = INT_MAX / 2;

int main() {
	int N; cin >> N;
	string s; cin >> s;
	vector<int> r(N), c(N);
	rep(i, N) scanf("%d", &r[i]);
	rep(i, N) scanf("%d", &c[i]);
	vector<int> diff(N + 1);
	int x = 0;
	ll ans = 0;
	rep(i, N) {
		x = (x + diff[i]) % 2;
		if (x != s[i] - '0') {
			ans += c[i];
			x = !x;
			diff[r[i]]++;
		}
	}
	cout << ans << endl;
}

atlasworld's blog