Submission #386210

#	Time	Username	Problem	Language	Result	Execution time	Memory
386210		Kalashnikov	Simple game (IZhO17_game)	C++17	100 / 100	101 ms	11248 KiB

game

#include <bits/stdc++.h>
 
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second
 
using namespace std;
using ll = long long;
 #define int ll
const int N = 1e6+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7 , P = 6547;

int a[N] , f[N];

void upd(int pos , int val) {
	for(; pos < N; pos |= pos+1) {
		f[pos] += val;
	}
}

int get(int x) {
	int res = 0;
	for(; x > 0; x &= x+1 , x--) {
		res += f[x];
	}
	return res;
}

void solve() {
	int n , m;
	cin >> n >> m;
	for(int i = 1; i <= n; i ++) {
		cin >> a[i];
		if(i != 1) {
			upd(min(a[i] , a[i-1]) , 1);
			upd(max(a[i] , a[i-1]) , -1);
		}
	}
	while(m --) {
		int t;
		cin >> t;
		if(t == 1) {
			int i , val;
			cin >> i >> val;
			if(i != 1) {
				upd(min(a[i] , a[i-1]) , -1);
				upd(max(a[i] , a[i-1]) , 1);
			}
			if(i != n) {
				upd(min(a[i] , a[i+1]) , -1);
				upd(max(a[i] , a[i+1]) , 1);
			}
			a[i] = val;
			if(i != 1) {
				upd(min(a[i] , a[i-1]) , 1);
				upd(max(a[i] , a[i-1]) , -1);
			}
			if(i != n) {
				upd(min(a[i] , a[i+1]) , 1);
				upd(max(a[i] , a[i+1]) , -1);
			}
		}
		else {
			int x;
			cin >> x;
			cout << get(x) << '\n';
		}
	}
	
}
/*
*/
main() {
    ios;
    int tt = 1;
    //cin >> tt;
    while(tt --) {
        solve();
    }
    return 0;
}

Compilation message (stderr)

game.cpp:75:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   75 | main() {
      |      ^

• Subtask 1: 22 / 22
• Subtask 2: 27 / 27
• Subtask 3: 51 / 51