제출 #762344

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
762344		vjudge1	새로운 문제 (POI11_met)	C++17	74 / 100	2636 ms	46928 KiB

met

#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;
 
const int N = 3e5+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7 , P = 6547;	

int p[N] , n , m , k , L[N] , R[N] , res[N];
ll t[N*4]; 
int l[N] , r[N] , c[N];
vector<int> g[N] , q[N] , q1[N];
void build(int v , int tl , int tr){
	t[v] = 0;
	if(tl == tr) return;
	int tm = (tl+tr) >> 1;
	build(v*2,tl,tm);
	build(v*2+1,tm+1,tr);
}
void upd(int v, int tl , int tr , int l , int r , int val){
	if(l <= tl && tr <= r){
		t[v] += val;
		return;
	}
	if(tl > r || tr < l) return;
	int tm = (tl+tr) >> 1;
	upd(v*2,tl,tm,l,r,val);
	upd(v*2+1,tm+1,tr,l,r,val);
}
ll get(int v, int tl , int tr , int pos){
	if(tl == tr) return t[v];
	int tm = (tl+tr) >> 1;
	if(tm >= pos) return get(v*2,tl,tm,pos)+t[v];
	else return get(v*2+1,tm+1,tr,pos)+t[v];
}
void solve(){
	cin >> n >> m;
	for(int i = 1; i <= m; i++) {
		int a;
		cin >> a;
		g[a].push_back(i);
	}
	for(int i = 1; i <= n; i++) cin >> p[i];
	cin >> k;
	for(int i = 1; i <= n; i++) {
		L[i] = 1;
		R[i] = k;
		q[(L[i]+R[i])/2].push_back(i);
	}
	for(int i = 1; i <= k; i++) cin >> l[i] >> r[i] >> c[i];
	int ok = 1;
	while(ok){
		ok = 0;
		build(1,1,m);
		for(int i = 1; i <= k; i++){
			if(l[i] > r[i]){
				upd(1,1,m,l[i],m,c[i]);
				upd(1,1,m,1,r[i],c[i]);
			} else {
				upd(1,1,m,l[i],r[i],c[i]);
			}
			for(int x : q[i]){
				ll sum = p[x];
				for(int y : g[x]) sum -= get(1,1,m,y);
				if(sum <= 0){
					res[x] = i;
					R[x] = i-1;
				} else {
					L[x] = i+1; 
				}
				if(L[x]  <= R[x]){
					q1[(L[x]+R[x])/2].push_back(x);
					ok++;
				}
			}
			q[i].clear();
		}
		swap(q,q1);
	}
	for(int i = 1; i <= n; i++) {
		if(res[i] == 0){
			cout << "NIE\n";
		} else {
			cout << res[i] << "\n";
		}
	}
}
/*

*/
signed main(){
	ios;
	solve();
	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