102292024-03-29 16:57:52111Egyengetőcpp17Időlimit túllépés 60/1003.549s33228 KiB
#include <bits/stdc++.h>
using namespace std;

#include "grader.h"

#define int long long

struct Nd{
	int a,d,l,r;
};

struct ST{
	int n,s;
	int root;
	Nd*v=new Nd[1000000];
	
	ST(int n):n(n),root(0),s(0){
	}
	
	void update(int&i,int l,int r,int ll,int rr){
		if(l>rr||r<ll){
			return;
		}
		if(!i){
			i=++s;
		}
		if(l>=ll&&r<=rr){
			v[i].a+=l-ll;
			v[i].d+=1;
			return;
		}
		update(v[i].l,l,(l+r)/2,ll,rr);
		update(v[i].r,(l+r)/2+1,r,ll,rr);
	}
	
	void update(int ll,int rr){
		update(root,0,n-1,ll,rr);
	}
	
	int query(int&i,int l,int r,int ii){
		if(!i||l>ii||r<ii){
			return 0;
		}
		int ans=v[i].a+v[i].d*(ii-l);
		if(l==r){
			return ans;
		}
		if(ii<=(l+r)/2){
			ans+=query(v[i].l,l,(l+r)/2,ii);
		}
		else{
			ans+=query(v[i].r,(l+r)/2+1,r,ii);
		}
		return ans;
	}
	
	int query(int ii){
		return query(root,0,n-1,ii);
	}
};

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int N=getN(),K=getK(),M=4e8;
	ST a(M+1),b(M+1);
	auto calc=[&](int x)->int{
		return a.query(x)+b.query(M/2-(x+K));
	};
	int pi=-1;
	for(int i=0;i<N;i++){
		int y=Data();
		a.update(y,M);
		b.update(M/2-y,M);
		int l=0,h=1e8;
		if(pi!=-1){
			l=min(pi,y),h=max(pi,y);
		}
		while(h-l>=3){
			int m1=l+(h-l)/3;
			int m2=h-(h-l)/3;
			if(calc(m1)<=calc(m2)){
				h=m2;
			}
			else{
				l=m1;
			}
		}
		int ans=1e18;
		for(int i=l;i<=h;i++){
			int a=calc(i);
			if(a<ans){
				ans=a;
				pi=i;
			}
		}
		Solution(ans);
	}
	return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
subtask10/0
1Elfogadva3ms1896 KiB
2Elfogadva3ms2096 KiB
subtask220/20
3Elfogadva4ms2380 KiB
4Elfogadva7ms2580 KiB
5Elfogadva8ms2580 KiB
6Elfogadva10ms2864 KiB
7Elfogadva13ms2760 KiB
8Elfogadva2.323s5096 KiB
subtask320/20
9Elfogadva16ms3188 KiB
10Elfogadva26ms4472 KiB
11Elfogadva25ms3892 KiB
12Elfogadva25ms3984 KiB
13Elfogadva25ms4328 KiB
14Elfogadva18ms4016 KiB
subtask420/20
15Elfogadva104ms3804 KiB
16Elfogadva270ms15060 KiB
17Elfogadva130ms4040 KiB
18Elfogadva279ms15280 KiB
19Elfogadva233ms5856 KiB
20Elfogadva264ms9116 KiB
subtask50/40
21Időlimit túllépés3.536s32568 KiB
22Elfogadva1.776s4596 KiB
23Elfogadva2.72s4412 KiB
24Elfogadva2.411s4192 KiB
25Időlimit túllépés3.549s32956 KiB
26Időlimit túllépés3.546s33228 KiB