Submission #1148260

#	Time	Username	Problem	Language	Result	Execution time	Memory
1148260		gazizmadi11	Meteors (POI11_met)	C++20	74 / 100	3302 ms	77104 KiB

met

//gm  --- akezhon
#include <bits/stdc++.h>
// #pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define pb push_back
#define pf push_front
#define F first
#define S second
#define all(v) v.begin(),v.end()
#define pii pair<int,int>
#define tm (tl+tr)/2
#define TL v+v, tl, tm
#define TR v+v+1, tm+1, tr
#define DA l <= tl && tr <= r
#define NE r < tl || tr < l
#define double long double
#define int long long 
using namespace std;

const int N=2e6+7;
const int mod=998244353;
// const int inf=2e18;

int n, m, k;
int l[N], r[N], a[N], o[N], w[N];
int L[N], R[N], ans[N], temp[N];
int t[N*4], p[N*4];
vector<int>pos[N];

void build(int v=1, int tl=1, int tr=m){
	t[v]=0;
	if(tl != tr){
		build(TL);
		build(TR);
	}
}
void upd(int l, int r, int val, int v=1, int tl=1, int tr=m){
	if(DA){
		t[v] += val;
		return;
	}
	if(NE)return;
	upd(l, r, val, TL);
	upd(l, r, val, TR);
}
int get(int pos, int v=1, int tl=1, int tr=m){
	if(tl==tr)return t[v];
	else if(pos <= tm)return t[v] + get(pos, TL);
	else return t[v] + get(pos, TR);
}
void add(int ind){
	if(l[ind] <= r[ind]){
		upd(l[ind], r[ind], a[ind]);
	}
	else{
		upd(l[ind], m, a[ind]);
		upd(1, r[ind], a[ind]);
	}
}

void AlemAmenov(){
	cin >> n >> m;
	for(int i=1; i <= m; i++){
		cin >> o[i];
		pos[o[i]].pb(i);
	}
	for(int i=1; i <= n; i++){
		cin >> w[i];
	}
	cin >> k;
	for(int i=1; i <= k; i++){
		cin >> l[i] >> r[i] >> a[i];
	}
	for(int i=1; i <= n; i++){
		L[i]=1;
		R[i]=k;
	}
	for(int op=1; op < 20; op++){
		vector<pii>v;
		for(int i=1; i <= n; i++){
			if(L[i] <= R[i])v.pb({(L[i]+R[i])/2, i});
		}
		sort(all(v));
		int cur=1;
		build();
		for(auto [m, ind] : v){
			while(cur <= m)add(cur++);
			int s=0;
			for(int p : pos[ind])s += get(p);
			if(w[ind] <= s){
				R[ind] = m-1;
				ans[ind] = m;
			}else{
				L[ind] = m+1;
			}
		}
	}
	for(int i=1; i <= n; i++){
		if(ans[i])cout << ans[i] << '\n';
		else cout << "NIE\n";
	}
}
signed main(){

    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int RealName=1;
    // cin >> RealName;

    // freopen("connect.in", "r", stdin);
    // freopen("connect.out", "w", stdout);

    while(RealName--)
        AlemAmenov();
    
return 0;
}

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