Submission #501752

#	Time	Username	Problem	Language	Result	Execution time	Memory
501752		Nuraly_Serikbay	Meteors (POI11_met)	C++14	0 / 100	311 ms	65536 KiB

met

#include <bits/stdc++.h>
  
#define endl "\n"
#define mp make_pair
#define pb push_back
#define F first
#define S second
#define sz size()
#define rep(i,k,n) for(int i = k ; i <= n ; ++i)
#define per(i,k,n) for(int i = k ; i >= n ; --i)
#define YOSIK() ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define all(x) x.begin(),x.end()
#define fr(x) freopen(x".in","r",stdin),freopen(x".out","w",stdout)
#define toqta return 0
#define PERMUTE next_permutation
#define no cout<<"No"<<endl;
#define yes cout<<"Yes"<<endl;
#define a() a + 1, a + n + 1  
#define int long long
  
using namespace std;

/*
#pragma GCC optimize("inline")
#pragma GCC optimize("-fgcse,-fgcse-lm")
#pragma GCC optimize("-ftree-pre,-ftree-vrp")
#pragma GCC optimize("-ffast-math")
#pragma GCC optimize("-fipa-sra")
#pragma GCC optimize("-fpeephole2")
#pragma GCC optimize("-fsched-spec")
#pragma GCC optimize("Ofast,no-stack-protector")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
	#pragma GCC optimize("unroll-loops")
*/

typedef long long ll;
typedef unsigned long long ull;
typedef string S;
typedef double D;
typedef long double lld;
   
const int N = 1e6 + 17;
const int modd = 1e9 + 7;
const ll INF = 1e18 - 19;
const int P = 37;
const ll NN = 1e7 + 17;
const D eps = 1e-19;
const double pi = 3.141592653589793238462643383279 ;   
 
int n, m;
int p[N], a[N], t[N], L[N], R[N], x[N];
vector <int> v[N];

void push (int v, int l, int r) {
	if (t[v] != 0 && l != r && v * 2 + 1 < N) {
		t[v * 2] += t[v];
		t[v * 2 + 1] += t[v];
		t[v] = 0;	
	}
}

void update (int v, int l, int r, int L, int R, int val) {	
	if (R < l || r < L) return;
	if (L <= l && r <= R) {
		t[v] += val;
		push(v, l, r);	
		return;
	}
	push(v, l, r);
	int mid = (l + r) / 2;
	update (v * 2, l, mid, L, R, val);
	update (v * 2 + 1, mid + 1, r, L, R, val);
}

int get (int v, int l, int r, int pos) {
	if (l == r) {	
		return t[v];
	}
	push(v, l, r);
	int mid = (l + r) / 2;
	if (pos <= mid) get (v * 2, l, mid, pos);
	else get (v * 2 + 1, mid + 1, r, pos);
}


int check (int w, int pos) {	
	for (int i = 1; i <= w; ++i) {
		if (L[i] > R[i]) {
			update(1, 1, m, L[i], m, x[i]);
			update(1, 1, m, 1, R[i], x[i]);
		} else {
			update(1, 1, m, L[i], R[i], x[i]);
		}
	}
	int ans = 0; 
	for (auto to : v[pos]) {
		ans += get(1, 1, m, to);	
	}
	for (int i = 1; i <= m * 4; ++i) t[i] = 0;
	return ans;

}

void Solution (/*Nuraly.Serikbay*/) {
	cin >> n >> m;
	for (int i = 1; i <= m; ++i) {
		cin >> p[i];
		v[p[i]].pb(i);	
	}
	for (int i = 1; i <= n; ++i) {
		cin >> a[i];	
	}
	int q;
	cin >> q;
	for (int i = 1; i <= q; ++i) {
		cin >> L[i] >> R[i] >> x[i];		 
	}
	for (int i = 1; i <= n; ++i) {
		int l = 0, r = q + 1;
		while (r - l > 1) {	
			int mid = (l + r) / 2;
			if (check(mid, i) >= a[i]) r = mid;
			else l = mid;
		}
		if (r == q + 1) cout << "NIE" << endl;
		else cout << r << endl;
	}
	
}
main () {
	YOSIK();
	ll T = 1;
	//cin >> T;
	while (T --) {
		Solution ();
	}
	exit (0);
}/*

*/

Compilation message (stderr)

met.cpp:130:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  130 | main () {
      | ^~~~
met.cpp: In function 'long long int get(long long int, long long int, long long int, long long int)':
met.cpp:83:1: warning: control reaches end of non-void function [-Wreturn-type]
   83 | }
      | ^

• Subtask 1: 0 / 12
• Subtask 2: 0 / 12
• Subtask 3: 0 / 12
• Subtask 4: 0 / 12
• Subtask 5: 0 / 13
• Subtask 6: 0 / 13
• Subtask 7: 0 / 13
• Subtask 8: 0 / 13