oj.uz

답안 #922968

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
922968 2024-02-06T11:20:21 Z vjudge1 Meteors (POI11_met) C++17
74 / 100
 1579 ms 65536 KB 
#include <bits/stdc++.h>
#define speed 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(x) (x).begin(), (x).end()
#define pb push_back
#define sz(x) (int)x.size()
#define F first
#define S second
#define int long long

const int dx[] = {0, 1, 0, -1, -1, 1, 1, -1};
const int dy[] = {1, 0, -1, 0, -1, 1, -1, 1};
const int maxn = 4e5+7;
const int inf = 1e9;
const int mod = 1e9+7;

using namespace std;

int n, m, k;
int s[maxn], need[maxn], ans[maxn];
int ql[maxn], qr[maxn], qx[maxn];
vector<int> cur[maxn], g[maxn];
int t[4 * maxn], p[4 * maxn];

struct node{
	int l, r;
} d[maxn];

void push(int tl, int tr, int v){
	if(tl != tr){
		p[v * 2] += p[v];
		p[v * 2 + 1] += p[v];
		p[v * 2] = min(p[v * 2], (int)1e9);
		p[v * 2 + 1] = min(p[v * 2 + 1], (int)1e9);
	}
	t[v] += p[v];
	t[v] = min(t[v], (int)1e9);
	p[v] = 0;
}

void build(int tl = 1, int tr = maxn, int v = 1){
	t[v] = p[v] = 0;
	if(tl == tr) return;
	int md = (tl + tr) / 2;
	build(tl, md, v * 2);
	build(md + 1, tr, v * 2 + 1);
}

void update(int l, int r, int val, int tl = 1, int tr = maxn, int v = 1){
	push(tl, tr, v);
	if(l > tr || tl > r) return;
	if(l <= tl && tr <= r){
		p[v] += val;
		p[v] = min(p[v], (int)1e9);
		push(tl, tr, v);
		return;
	}
	int md = (tl + tr) / 2;
	update(l, r, val, tl, md, v * 2);
	update(l, r, val, md + 1, tr, v * 2 + 1);
}

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

void solve(){
	cin >> n >> m;
	for(int i = 1; i <= m; i++){
		cin >> s[i];
		g[s[i]].push_back(i);
	}
	for(int i = 1; i <= n; i++) cin >> need[i];
	cin >> k;
	for(int i = 1; i <= n; i++) d[i] = {1, k};
	for(int i = 1; i <= k; i++) cin >> ql[i] >> qr[i] >> qx[i];
	for(int rep = 1; rep <= 25; rep++){
		for(int i = 1; i <= n; i++){
			if(d[i].l > d[i].r) continue;
			int mid = (d[i].l + d[i].r) / 2;
			cur[mid].pb(i);
		}
		build();
		for(int mid = 1; mid <= k; mid++){
			if(ql[mid] <= qr[mid]){
				update(ql[mid], qr[mid], qx[mid]);
			}else{
				update(ql[mid], m, qx[mid]);
				update(1ll, qr[mid], qx[mid]);
			}
			if(!cur[mid].size()) continue;
			for(auto x : cur[mid]){
				int sum = 0;
				for(auto pos : g[x]){
					sum += get(pos);
				}
				if(sum >= need[x]){
					ans[x] = mid;
					d[x].r = mid - 1;
				}else{
					d[x].l = mid + 1;
				}
			}
		}
		for(int i = 1; i <= k; i++){
			cur[i].clear();
		}
	}
	for(int i = 1; i <= n; i++){
		if(ans[i] == 0) cout << "NIE\n";
		else cout << ans[i] << '\n';
	}
}

signed main(){
	speed
	int test = 1;
//	cin >> test;
	while(test--){
		solve();
	}
}

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 55 ms 51544 KB Output is correct
2 Correct 46 ms 51800 KB Output is correct
3 Correct 48 ms 51792 KB Output is correct

Subtask #2
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 49 ms 51804 KB Output is correct
2 Correct 54 ms 51800 KB Output is correct
3 Correct 50 ms 51804 KB Output is correct

Subtask #3
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 693 ms 57312 KB Output is correct
2 Correct 935 ms 62232 KB Output is correct
3 Correct 807 ms 61464 KB Output is correct

Subtask #4
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 840 ms 58660 KB Output is correct
2 Correct 857 ms 58756 KB Output is correct
3 Correct 966 ms 62556 KB Output is correct
4 Correct 161 ms 58672 KB Output is correct

Subtask #5
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 393 ms 57620 KB Output is correct
2 Correct 659 ms 64808 KB Output is correct
3 Correct 560 ms 53872 KB Output is correct
4 Correct 788 ms 61944 KB Output is correct

Subtask #6
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 896 ms 56448 KB Output is correct
2 Correct 843 ms 58416 KB Output is correct
3 Correct 728 ms 56912 KB Output is correct
4 Correct 935 ms 63784 KB Output is correct

Subtask #7
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 1113 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 1579 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -