zoukankan      html  css  js  c++  java
  • CF1500E Subset Trick

    XLVII.CF1500E Subset Trick

    考虑对于每个集合大小 \(i\),找到所有大小为 \(i\) 的集合中元素和最小的一个 \(l_i\) 与最大的一个 \(r_i\)。则,所有 \(x\in[l_i,r_i)\) 均不合法。

    于是我们就要求 \(\Big|\bigcup\limits_{i=0}^{n}[l_i,r_i)\Big|\)

    直接求不容易;但是我们可以正难则反,考虑找出那些不被包含在其中的 \(x\)。虽然这一 \(x\) 的数量是无限的,但是发现若设所有数之和为 \(s\),则所有 \(x\in[s,\infty)\) 显然均不被包含,可以不考虑。于是我们仅需考虑那些 \(\in[0,s)\) 且不被包含的数即可。

    显然的是,\(l_i\) 必然是前 \(i\) 小的数之和,\(r_i\) 必然是前 \(i\) 大的数之和。

    假如我们在平面直角坐标系中画出所有 \((i,l_i)\)\((i,r_i)\) 的话,就会发现所有 \(l\) 连成的折线是凹的,所有 \(r\) 连成的这些是凸的,且这两条折线是中心对称的。

    于是我们仅需考虑前一半中 \(x\) 的数量,然后乘二就得到了全体 \(x\) 的数量(当然,若 \(x\) 是奇数,还要额外考虑中间的那个位置,但是随便特判一下就出来了)。

    现在考虑每个不被包含在其中的 \(x\) 的区间,发现其都是形如 \([r_{i-1},l_i)\) 的形式。

    一个非常棒的观察是,因为 \(l\)\(r\) 凸且二者对称,所以在前一半中 \(r\) 的增量总是不小于 \(l\) 的增量,这也意味着上述区间若非空则所有这样的 \(i\) 是一段前缀。于是我们可以用什么数据结构维护一下 \(l\)\(r\),然后套个二分就能找到最大的这样的 \(i\)

    现在考虑用什么数据结构。平衡树显然是可行的,但是据说被卡常了。于是离散化后上线段树维护即可。

    时间复杂度 \(O(n\log^2n)\)。那个二分不能放到线段树上实现,因为要维护的是两棵不同的线段树。

    代码:

    #include<bits/stdc++.h>
    using namespace std;
    typedef long long ll;
    int n,m,lim,sz;
    ll a[200100],b[200100],sum;
    vector<ll>v;
    struct ST{
    	#define lson x<<1
    	#define rson x<<1|1
    	#define mid ((l+r)>>1)
    	struct SegTree{int sz;ll sum,tag;}seg[1600100];
    	void ADD(int x,ll y){seg[x].tag+=y,seg[x].sum+=seg[x].sz*y;}
    	void pushdown(int x){ADD(lson,seg[x].tag),ADD(rson,seg[x].tag),seg[x].tag=0;}
    	void pushup(int x){seg[x].sum=seg[lson].sum+seg[rson].sum,seg[x].sz=seg[lson].sz+seg[rson].sz;};
    	void turnon(int x,int l,int r,int P){if(l>P||r<P)return;if(l==r)seg[x].sum=seg[x].tag,seg[x].sz=1;else pushdown(x),turnon(lson,l,mid,P),turnon(rson,mid+1,r,P),pushup(x);}
    	void turnoff(int x,int l,int r,int P){if(l>P||r<P)return;if(l==r)seg[x].sum=0,seg[x].sz=0;else pushdown(x),turnoff(lson,l,mid,P),turnoff(rson,mid+1,r,P),pushup(x);}
    	void modify(int x,int l,int r,int L,int R,ll val){if(l>R||r<L)return;if(L<=l&&r<=R)ADD(x,val);else pushdown(x),modify(lson,l,mid,L,R,val),modify(rson,mid+1,r,L,R,val),pushup(x);}
    	ll kth(int x,int l,int r,int k){if(l==r)return seg[x].sum;pushdown(x);if(seg[lson].sz>=k)return kth(lson,l,mid,k);else return kth(rson,mid+1,r,k-seg[lson].sz);}
    	ll query(int x,int l,int r,int k){if(l==r)return seg[x].sum;pushdown(x);if(seg[lson].sz>=k)return query(lson,l,mid,k);return seg[lson].sum+query(rson,mid+1,r,k-seg[lson].sz);}
    	void iterate(int x,int l,int r){printf("%d[%d,%d]:%d,%d\n",x,l,r,seg[x].sum,seg[x].sz);if(l!=r)pushdown(x),iterate(lson,l,mid),iterate(rson,mid+1,r);}
    }s1,s2;//s1 is minimum,s2 is maximum
    void turnon(int P){s1.turnon(1,1,lim,P),s1.modify(1,1,lim,P,lim,v[P-1]);s2.turnon(1,1,lim,lim-P+1),s2.modify(1,1,lim,lim-P+1,lim,v[P-1]);sz++,sum+=v[P-1];}
    void turnoff(int P){s1.turnoff(1,1,lim,P),s1.modify(1,1,lim,P,lim,-v[P-1]);s2.turnoff(1,1,lim,lim-P+1),s2.modify(1,1,lim,lim-P+1,lim,-v[P-1]);sz--,sum-=v[P-1];}
    ll query(){
    	if(sz<=1)return 0;
    	int l=1,r=sz>>1;
    	while(l<r){
    		int md=(l+r+1)>>1;
    		if(s1.kth(1,1,lim,md)>=s2.kth(1,1,lim,md-1))l=md;
    		else r=md-1;
    	}
    	ll ret=s1.query(1,1,lim,l);if(l>1)ret-=s2.query(1,1,lim,l-1);
    	ret<<=1;
    	if(sz&1)ret+=max(0ll,s1.kth(1,1,lim,(sz+1)>>1)-s2.kth(1,1,lim,sz>>1));
    //	printf("(%d,%d)\n",sum,sz);
    	return sum-ret;
    }
    int main(){
    	scanf("%d%d",&n,&m);
    	for(int i=1;i<=n;i++)scanf("%lld",&a[i]),v.push_back(a[i]);
    	for(int i=1,x;i<=m;i++)scanf("%d%lld",&x,&b[i]),v.push_back(b[i]),b[i]=(x==1?b[i]:-b[i]);
    	sort(v.begin(),v.end()),v.resize(lim=unique(v.begin(),v.end())-v.begin());
    	for(int i=1;i<=n;i++)a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin()+1;
    	for(int i=1;i<=m;i++){
    		if(b[i]>0)b[i]=lower_bound(v.begin(),v.end(),b[i])-v.begin()+1;
    		else b[i]=-(lower_bound(v.begin(),v.end(),-b[i])-v.begin()+1);
    	}
    	for(int i=1;i<=n;i++)turnon(a[i]);
    	printf("%lld\n",query());
    	for(int i=1;i<=m;i++){
    		if(b[i]>0)turnon(b[i]);
    		else turnoff(-b[i]);
    //		s1.iterate(1,1,lim),puts("");
    //		s2.iterate(1,1,lim),puts("");
    		printf("%lld\n",query());
    	}
    	return 0;
    }
    
  • 相关阅读:
    内部类
    多重继承关系初始化顺序及初始化
    String
    Error
    算法:插入排序
    算法:冒泡排序
    算法:选择排序
    注册Activity
    java变量的作用域和基本数据类型转换
    java数据类型
  • 原文地址:https://www.cnblogs.com/Troverld/p/14613591.html
Copyright © 2011-2022 走看看