zoukankan      html  css  js  c++  java
  • BZOJ.5319.[JSOI2018]军训列队(主席树)

    LOJ
    BZOJ
    洛谷

    看错了,果然不是(ZJOI)。。(jry)(JSOI)出这么水的题做T3么= =
    感觉说的有点乱,不要看我写的惹=-=


    对于询问(l,r,k),设(t=r-l+1)。对于指定区间([k,k+t-1]),显然(k)左边的人都要从(k)开始依次排列,(k+t-1)右边的人要从(k+t-1)往左依次排列。区间中的比较麻烦。
    也比较显然的是存在一个点(midin[k-1,k+t-1]),使得(mid)左边的人都被排在([k,mid])中,右边的人都被排到([mid+1,k+t-1])中。而且代价很好算,就是需要求一个等差数列的和,以及左边所有人的坐标和。
    考虑如何找这个(mid)。显然有(mid)左边人的数量等于(mid-k+1),所以可以二分+主席树,但是是(log^2)的。
    注意到查的区间是个前缀,所以可以直接在主席树上二分。
    那么复杂度就是(O(nlog n))啦。

    莫名想写指针
    为啥我的主席树这么慢_(:з」∠)_


    //210300kb	12428ms
    #include <cstdio>
    #include <cctype>
    #include <algorithm>
    #define F(l,r) (1ll*((r)-(l)+1)*(l+r)>>1)//n(a1+an)/2... l<=r+1
    #define gc() getchar()
    #define MAXIN 500000
    //#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
    typedef long long LL;
    const int N=5e5+5;
    
    int A[N],root[N];
    char IN[MAXIN],*SS=IN,*TT=IN;
    struct Segment_Tree
    {
    	#define ls son[x][0]
    	#define rs son[x][1]
    	#define lson ls,l,m,
    	#define rson rs,m+1,r
    	#define lson2 ls,son[y][0],l,m
    	#define rson2 rs,son[y][1],m+1,r
    	#define S N*21
    	int tot,son[S][2],sz[S];
    	LL lsum,rsum,sum[S];
    	#undef S
    	void Insert(int &x,int y,int l,int r,int p)
    	{
    		sz[x=++tot]=sz[y]+1, sum[x]=sum[y]+p;
    		if(l==r) return;
    		int m=l+r>>1;
    		p<=m?(rs=son[y][1],Insert(lson2,p)):(ls=son[y][0],Insert(rson2,p));
    	}
    	int Query(int x,int y,int l,int r,int K)//y-x
    	{
    		if(l==r) return lsum+=sum[y]-sum[x],l;//mid=k-1 -> lsum=sum[now]=0
    		int m=l+r>>1,t=sz[son[y][0]]-sz[ls];
    		return m-K+1>=t?(rsum+=sum[son[y][1]]-sum[rs],Query(lson2,K)):(lsum+=sum[son[y][0]]-sum[ls],Query(rson2,K+t));
    	}
    }T;
    
    inline int read()
    {
    	int now=0;register char c=gc();
    	for(;!isdigit(c);c=gc());
    	for(;isdigit(c);now=now*10+c-48,c=gc());
    	return now;
    }
    
    int main()
    {
    	int n=read(),Q=read(),mx=0,mn=N;
    	for(int *s=A+1,*ed=A+1+n; s!=ed; ++s) mx=std::max(mx,*s=read()), mn=std::min(mn,*s);
    	for(int i=1; i<=n; ++i) T.Insert(root[i],root[i-1],mn,mx,A[i]);
    	for(int l,r,ql,qr; Q--; )
    	{
    		l=read(),r=read(),ql=read(),qr=ql+r-l;
    		if(ql<=mn||qr>=mx) printf("%lld
    ",std::abs(T.sum[root[r]]-T.sum[root[l-1]]-F(ql,qr)));
    		else
    		{
    			T.lsum=0, T.rsum=0;
    			int mid=T.Query(root[l-1],root[r],mn,mx,ql);
    			printf("%lld
    ",F(ql,mid)-T.lsum+T.rsum-F(mid+1,qr));
    		}
    	}
    	return 0;
    }
    
  • 相关阅读:
    时间,关机重启及网络常识
    bash xshell 特性
    Http介绍
    rsync 守护进程模式小记
    定时任务+邮件发送 小记
    4.iptables的匹配条件(一)
    3.iptables规则管理
    2.iptables规则查询
    1.iptables概念
    4.LVS实验构建
  • 原文地址:https://www.cnblogs.com/SovietPower/p/10718566.html
Copyright © 2011-2022 走看看