Submission #15404

#	Time	Username	Problem	Language	Result	Execution time	Memory
15404		gs14004	통로 위의 개미 (kriii3_X)	C++14	30 / 85	3000 ms	6480 KiB

X

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long lint;

lint l;
int q;

int ants_kth[200005];

struct node{
	node *ls, *rs;
	int v;
	node(){
		ls = rs = NULL;
		v = 0;
	}
};

void add(lint pos, lint ps, lint pe, node* p){
	p->v++;
	if(ps == pe) return;
	lint pm = (ps + pe) / 2;
	if(pos <= pm){
		if(!p->ls) p->ls = new node();
		add(pos, ps, pm, p->ls);
	}
	else{
		if(!p->rs) p->rs = new node();
		add(pos, pm + 1, pe, p->rs);
	}
}

int sum(lint s, lint e, lint ps, lint pe, node* p){
	if(e < ps || pe < s) return 0;
	if(s <= ps && pe <= e) return p->v;;
	lint pm = (ps + pe) / 2;
	int ret = 0;
	if(p->ls) ret += sum(s,e,ps,pm,p->ls);
	if(p->rs) ret += sum(s,e,pm+1,pe,p->rs);
	return ret;
}

node *root1, *root2;

inline int get_sum(lint s, lint e, node *r){
	if(e >= 2 * l){
		return sum(s, 2*l-1, 0, 2*l-1, r) + sum(0, e - 2*l, 0, 2*l-1, r);
	}
	return sum(s, e, 0, 2*l-1, r);
}

inline lint norm(lint x){
	x %= 2 * l;
	x += 2 * l;
	x %= 2 * l;
	return x;
}

	int piv = 1;

int get_low(lint p, lint t){
	if(p == l) return piv-1;
	int ret = 0;
	ret += get_sum(norm(-t), norm(-t) + p, root1);
	ret += get_sum(norm(2 * l - p - t), norm(2 * l - p - t) + p - 1, root1);
	ret += get_sum(norm(t), norm(t) + p, root2);
	ret += get_sum(norm(2 * l - p + t), norm(2 * l - p + t) + p - 1, root2);
	return ret;
}

int main(){
	root1 = new node();
	root2 = new node();
	scanf("%lld %d",&l,&q);
	for(int i=0; i<q; i++){
		lint t;
		lint p;
		scanf("%lld %lld",&t,&p);
		if(p == 1){
			lint pos, dx;
			scanf("%lld %lld",&pos, &dx);
			int k = 1 + get_low(pos,t);
			for(int i=1; i<piv; i++){
				if(ants_kth[i] >= k){
					ants_kth[i]++;
				}
			}
			ants_kth[piv++] = k;
			lint p = 1ll * pos - (1ll * dx * t % (2 * l)) + 2 * l;
			p %= (2 * l);
			p += 2 * l;
			p %= 2 * l;
			if(dx < 0) add(p,0,2*l-1,root2);
			else add(p,0,2*l-1,root1);
		}
		else{
			int x;
			scanf("%d",&x);
			x = ants_kth[x];
			int s = 0, e = l;
			while(s != e){
				int m = (s+e)/2;
				if(get_low(m,t) < x) s = m+1;
				else e = m;
			}
			printf("%d\n",s);
		}
	}
}

• Subtask 1: 30 / 30
• Subtask 2: 0 / 55