제출 #432135

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
432135		dreezy	휴가 (IOI14_holiday)	C++17	23 / 100	53 ms	9108 KiB

holiday

#include <bits/stdc++.h>
#include"holiday.h"
using namespace std;
#define ll long long
#define pill pair<int,ll>
#define pi pair<int,int>
#define f first
#define s second
/***********************/
struct SegTree{
	vector<pill> st;
	int sz;
	SegTree(int n){
		sz = n;
		st.assign(n*4, {0,0ll});
	}
	
	
	void updactive(int n, int l, int r, int ind, int val){
		if(l == r){
			st[n].f = val;
			return;
		}
		
		int mid = (l + r)/2;
		
		if(mid >= ind)
			updactive(2*n+1, l, mid, ind, val);
		else
			updactive(2*n + 2, mid + 1, r, ind, val);
		st[n].f = st[2*n+1].f + st[2*n+2].f;
	}
	
	void updval(int n, int l, int r, int ind, int val){
		if(l == r){
			st[n].s = val;
			return;
		}
		
		int mid = (l + r)/2;
		
		if(mid >= ind)
			updval(2*n+1, l, mid, ind, val);
		else
			updval(2*n + 2, mid + 1, r, ind, val);
		st[n].s = st[2*n+1].s + st[2*n+2].s;
	}
	

	void activate(int city,int val){
		updactive(0,0,sz, city, 1);
		updval(0,0,sz, city, val);
	}
	
	void deactivate(int city){
		updactive(0,0,sz, city, 0);
		updval(0,0,sz, city, 0);
	}
	
	int getfirst(int n, int l, int r, int val){
		if(l == r) return l;
		int mid = (l + r)/2;
		
		if(st[2*n+1].f >= val )
			return getfirst(2*n+1, l, mid, val);
		return getfirst(2*n+ 2, mid + 1, r, val - st[2*n+1].f);
	}
	
	int getfirst(int val){
		int target = st[0].f - val +1;
		if(target <=0)
			return 0;
		return getfirst(0, 0, sz, target);
	}
	
	
	ll qrysum(int n, int l, int r, int s, int e){
		if(s<= l and e >= r)	return st[n].s;
		
		ll ans = 0;
		int mid = (l + r)/2;
		
		if( mid >= s)
			ans += qrysum(2*n+1, l, mid, s, e);
		if(mid < e)
			ans+=qrysum(2*n+2, mid + 1, r, s,e);
		return ans;
	}
	
	ll getlargest(int rng){
		if(rng <= 0) return 0;
		int lower = getfirst(rng);
		//cout << lower<<endl;
		return qrysum(0, 0,sz, lower, sz );
		
	}

	
};



long long int findMaxAttraction(int n, int start, int d, int attraction[]) {
	ll ans = 0;
	
	SegTree st(n);
	//spend right days travelling
	//visit day - right cities
	
	vector<pi> cities(n);
	for(int i =0; i<n; i++){
		cities[i] = {attraction[i], i};
	}
	
	sort(cities.begin(), cities.end());
	

		
	vector<int> inds(n);
	for(int i= 0; i<n;i++)
		inds[cities[i].s] = i;
	
	for(int right = 0; right <n; right++){
		st.activate(inds[right], attraction[right]);
		
		//cout << right<<": "<< inds[right]<<", "<<st.getlargest(d - right)<<endl;
		ans = max(ans, st.getlargest(d - right));

	}
	
	
    return ans;
}
/***********************/

• Subtask 1: 0 / 7
• Subtask 2: 23 / 23
• Subtask 3: 0 / 17
• Subtask 4: 0 / 53