Submission #865726

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
865726	2023-10-24T14:48:47 Z	vjudge1	Meteors (POI11_met)	C++17	74 / 100	4650 ms	65536 KB

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <ctime>
#include <deque>
#include <queue>
#include <stack>
#include <string>
#include <bitset>
#include <cstdio>
#include <limits>
#include <vector>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <fstream>
#include <numeric>
#include <sstream>
#include <cassert>
#include <iomanip>
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <fstream>
#include <unordered_map>

using namespace std;

typedef long long ll;
typedef long double ld;

//#define int long long

#define all(x) x.begin(),x.end()
#define pb push_back
#define ent "\n"

const int maxn = (int)3e5 + 13;
const ll inf = (long long)1e18 + 20;
const int mod = (int)1e9 + 7;

int n,m;
vector<int>pos[maxn],MID[maxn];
int a[maxn];
int l[maxn],r[maxn],w[maxn];
pair<int,int>seg[maxn];
ll t[maxn * 4],q[maxn * 4];

void push(int v,int tl,int tr){
	if(q[v]){
		t[v] += q[v] * (tr - tl + 1);
		if(tl < tr){
			q[v * 2] += q[v];
			q[v * 2 + 1] += q[v];
		}
		q[v] = 0;
	}
}

void upd(int v,int tl,int tr,int l,int r,int x){
	push(v,tl,tr);
	if(l <= tl && tr <= r){
		q[v] += x;
		push(v,tl,tr);
		return;
	}
	int mid = (tl + tr) >> 1;
	if(l > tr || r < tl)return;
	upd(v * 2,tl,mid,l,r,x);
	upd(v * 2 + 1,mid + 1,tr,l,r,x);
	t[v] = t[v * 2] + t[v * 2 + 1];
}

ll get(int v,int tl,int tr,int pos){
	push(v,tl,tr);
	if(tl == tr){
		return t[v];
	}
	int mid = (tl + tr) >> 1;
	if(pos <= mid){
		return get(v * 2,tl,mid,pos);
	}
	else{
		return get(v * 2 + 1,mid + 1,tr,pos);
	}
}

void build(int v,int tl,int tr){
	if(tl == tr){
		t[v] = 0;
		q[v] = 0;
		return;
	}
	int mid = (tl + tr) >> 1;
	build(v * 2,tl,mid);
	build(v * 2 + 1,mid + 1,tr);
	t[v] = 0,q[v] = 0;
}

ll ans[maxn];

void solve(){
	cin >> n >> m;
	for(int i = 1 ; i <= m ; i ++){
		int o;cin >> o;
		pos[o].pb(i);
	}
	for(int i = 1 ; i <= n ; i ++){
		cin >> a[i];
	}
	int k;cin >> k;
	for(int i = 1 ; i <= k ; i++){
		cin >> l[i] >> r[i] >> w[i];
	}
	for(int i = 1 ; i <= n ; i ++){
		seg[i] = {1, k};
	}
	int ok = 1;
	while(ok){
		build(1,1,m), ok = 0;
		for(int i = 1 ; i <= n ; i ++){
			if(seg[i].first <= seg[i].second){
				int mid = (seg[i].first + seg[i].second) >> 1;
				MID[mid].pb(i);
				ok = 1;
			}
		}
		for(int i = 1 ; i <= k ; i ++){
			if(l[i] <= r[i]){
				upd(1,1,m,l[i],r[i],w[i]);
			}
			else{
				upd(1,1,m,l[i],m,w[i]);
				upd(1,1,m,1,r[i],w[i]);
			}
			ll sum = 0;
			for(int x : MID[i]){
				sum = 0;
				for(int y : pos[x]){
					sum += get(1,1,m,y);
					if(sum >= a[x])break;
				}
				if(sum >= a[x]){
					seg[x].second = i - 1;
					ans[x] = i;
				}
				else{
					seg[x].first = i + 1;
				}
			}
			MID[i].clear();
		}
	}
	for(int i = 1 ; i <= n ; i ++){
		if (ans[i] == 0) cout << "NIE\n";
		else cout << ans[i] << '\n';
	}
}

signed main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int t = 1;
	//cin >> t;
	while(t --){
		solve();
	}
	return 0;
}

Subtask #1
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	8 ms	25948 KB	Output is correct
2	Correct	9 ms	26172 KB	Output is correct
3	Correct	9 ms	25948 KB	Output is correct

Subtask #2
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	8 ms	25948 KB	Output is correct
2	Correct	8 ms	25948 KB	Output is correct
3	Correct	11 ms	26200 KB	Output is correct

Subtask #3
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	418 ms	28824 KB	Output is correct
2	Correct	510 ms	30800 KB	Output is correct
3	Correct	491 ms	30484 KB	Output is correct

Subtask #4
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	477 ms	30032 KB	Output is correct
2	Correct	500 ms	29828 KB	Output is correct
3	Correct	489 ms	32968 KB	Output is correct
4	Correct	74 ms	29780 KB	Output is correct

Subtask #5
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	216 ms	29196 KB	Output is correct
2	Correct	378 ms	33548 KB	Output is correct
3	Correct	240 ms	26428 KB	Output is correct
4	Correct	475 ms	30800 KB	Output is correct

Subtask #6
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	470 ms	28752 KB	Output is correct
2	Correct	420 ms	29776 KB	Output is correct
3	Correct	466 ms	28760 KB	Output is correct
4	Correct	512 ms	34200 KB	Output is correct

Subtask #7
0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4650 ms	55212 KB	Output is correct
2	Correct	2553 ms	41812 KB	Output is correct
3	Correct	1281 ms	26648 KB	Output is correct
4	Runtime error	4177 ms	65536 KB	Execution killed with signal 9
5	Halted	0 ms	0 KB	-

Subtask #8
0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4617 ms	53916 KB	Output is correct
2	Correct	1381 ms	42020 KB	Output is correct
3	Correct	1033 ms	25948 KB	Output is correct
4	Runtime error	3559 ms	65536 KB	Execution killed with signal 9
5	Halted	0 ms	0 KB	-

Submission #865726

Compilation message