【BZOJ】4908: [BeiJing2017]开车-分块

传送门：bzoj4908

---

题解

一个显然的贪心是按位置排序后第$k$大的车到第$k$大的加油站处加油。

但维护每个车的位置显然不好统计答案(因为每次修改相当于一段车整体移动，与加油站的相对位置无法统计)。

于是转换为离散化位置后统计每一个单位距离对于答案的贡献：
类似于括号匹配：每个车位置$+1$，加油站位置$-1$，设前缀和为$c_i$，则第$i$个单位距离的贡献为$|c_i|\times di{s}_i$(有$|c_i|$对车和加油站经过了这个单位距离)。

这个可以分块维护，每个快内维护有序数列，散块$O(\sqrt{n}\log \sqrt{n})$维护，整块$O(\log \sqrt{n})$维护。有点麻烦。

---

代码

#include<bits/stdc++.h>
using namespace std;
const int N=5e4+100;
typedef long long ll;

int n,m,pos[N],gas[N],rv[N*3],num,tot,cot;
int bs,bel[N*3],c[N*3],rk[N*3];ll ans;

inline bool cmp(const int&x,const int&y){return c[x]<c[y];}

struct blk{
    int val[405],ss[405],chg,st,ed,sz;ll cst;
    inline void reb(){
    	chg=0;cst=0LL;int i,j,k;sz=ed-st+1;
		for(i=st;i<=ed;++i) rk[i]=i;
		sort(rk+st,rk+ed+1,cmp);
		for(i=1;i<=sz;++i){
			k=rk[i+st-1];j=rv[k+1]-rv[k];
			ss[i]=ss[i-1]+j;val[i]=c[k];
			cst+=abs(val[i])*j;
		}
    }
    inline void ad(int v){
    	if(val[1]+chg>0){
    		cst+=ss[sz]*v;
    	}else if(val[sz]+chg<0){
    		cst+=ss[sz]*(-v);
    	}else{
    		int i,x,y;
    		x=lower_bound(val+1,val+sz+1,-chg)-val;
    		cst+=ss[x-1]*(-v);
    		if(val[x]+chg==0){
    			y=upper_bound(val+1,val+sz+1,-chg)-val-1;
    			cst+=(ss[y]-ss[x-1]);cst+=(ss[sz]-ss[y])*v;
    		}else cst+=(ss[sz]-ss[x-1])*v;	
    	}
		chg+=v;
    }
}b[405];
struct qr{int id,to;}q[N];
struct P{
	int op,id,v;
	bool operator<(const P&ky)const{
		return v<ky.v;
	}
}t[N*3];

inline void init()
{
	int i,j,x,y;
    scanf("%d",&n);
    for(i=1;i<=n;++i){scanf("%d",&x);t[i]=(P){0,i,x};}
    for(i=1;i<=n;++i){scanf("%d",&x);t[i+n]=(P){1,i,x};}
    tot=n+n;scanf("%d",&m);
    for(i=1;i<=m;++i){
    	scanf("%d%d",&x,&y);
    	t[++tot]=(P){2,i,y};q[i]=(qr){x,y};
    }
    sort(t+1,t+tot+1);
    for(i=1;i<=tot;++i){
    	if((i==1)||(t[i].v!=rv[num])) rv[++num]=t[i].v;
    	if(!t[i].op) pos[t[i].id]=num;
    	else if(t[i].op==1) gas[t[i].id]=num;
    	else q[t[i].id].to=num;
    }
    rv[num+1]=rv[num];
}

inline void build()
{
	int i,j;bs=(int)sqrt(num)+1;cot=(num-1)/bs+1;
	for(i=1;i<=cot;++i) b[i].st=b[i-1].ed+1,b[i].ed=b[i-1].ed+bs;
	b[cot].ed=num;
	for(i=1;i<=num;++i) bel[i]=(i-1)/bs+1;
	for(i=1;i<=n;++i) 
	 c[pos[i]]++,c[gas[i]]--;
	for(i=1;i<=num;++i) c[i]+=c[i-1];
	for(i=1;i<=cot;++i) {b[i].reb();ans+=b[i].cst;}
	printf("%lld\n",ans);
}

inline void modify(int x,int v)
{
	int i,bl=bel[x],l=b[bl].st,r=b[bl].ed;
	for(i=l;i<x;++i) c[i]+=b[bl].chg;
	for(i=x;i<=r;++i) c[i]+=b[bl].chg+v;
	ans-=b[bl].cst;b[bl].reb();ans+=b[bl].cst;
	for(bl++;bl<=cot;++bl) {ans-=b[bl].cst;b[bl].ad(v);ans+=b[bl].cst;}
}

inline void sol()
{
	int i,j;
	for(i=1;i<=m;++i){
	    modify(pos[q[i].id],-1);
	    pos[q[i].id]=q[i].to;
	    modify(q[i].to,1);
	    printf("%lld\n",ans);
	}
}

int main(){
	init();
	build();
	sol();
	return 0;
}