线段树提升练习题

CF283E. Cow Tennis Tournament （正难则反 + 线段树维护异或）

链接：https://codeforces.com/contest/283/problem/E

题意：有 $n$ 个能力值，能力值越大越强。有 $m$ 个区间，每个区间代表能力值在 $[a,b]$ 内的人的强弱关系翻转，区间外的人对区间内的人的强弱关系不变（没有传递关系）。问有多少个三元组 $(p,q,r)$满足：$p$ 打败 $q$，$q$ 打败 $r$，$r$ 打败 $p$ 。$(1\le n\le 10^5)$

思路：由于不可能存在 a 打败 b ，b 打败 a 的情况，那么不符合的情况一定是一个人可以打败两个人的情况。

• 因此可以从 $C_n^3$ 中减去不符合的情况。这样计算出每个人可以打败的总人数就可以了。
• 一个人可以打败的人数为两部分的和：1. 初始能力值小于自己且翻转偶数次的人数。2. 初始能力值大于自己且翻转奇数次的人数。
• 考虑一个人怎么算，我们只需要将所有包含他的区间的影响添加上去，不包含他的区间的影响取消掉，此时去查询就可以了。（而且只能是这样才是正确的，而且更新完之后，只有对他一个人是正确的）
• 考虑按能力值从小到大枚举每一个人，假设当前为第 $i$ 个人，那么区间左端点在 $[1,i]$ 的都必须更新，区间右端点在 $[1,i-1]$ 的都必须取消。这样就可以，递推地去更新了。

#include <bits/stdc++.h>
#define ls (rt<<1)
#define rs (rt<<1|1)
#define ll long long
using namespace std;
const int maxn=1e5+5;

int n,m,a[maxn];
vector<int> v1[maxn],v2[maxn];
int st[maxn<<2],lazy[maxn<<2];

void pushDown(int rt,int L,int R)
{
    if(lazy[rt])
    {
        int mid=(L+R)>>1;
        st[ls]=mid-L+1-st[ls];
        st[rs]=R-mid-st[rs];
        lazy[ls]^=1;
        lazy[rs]^=1;
        lazy[rt]=0;
    }
}
void update(int rt,int l,int r,int L,int R)
{
    if(l<=L&&R<=r)
    {
        st[rt]=R-L+1-st[rt];
        lazy[rt]^=1;
        return;
    }
    pushDown(rt,L,R);
    int mid=(L+R)>>1;
    if(l<=mid) update(ls,l,r,L,mid);
    if(r>mid) update(rs,l,r,mid+1,R);
    st[rt]=st[ls]+st[rs];
}
int query(int rt,int l,int r,int L,int R)
{
    if(l<=L&&R<=r) return st[rt];
    pushDown(rt,L,R);
    int mid=(L+R)>>1;
    int ans=0;
    if(l<=mid) ans+=query(ls,l,r,L,mid);
    if(r>mid) ans+=query(rs,l,r,mid+1,R);
    return ans;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; ++i) scanf("%d",&a[i]);
    sort(a+1,a+1+n);
    for(int i=1; i<=m; ++i)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        l=lower_bound(a+1,a+1+n,l)-a;
        r=upper_bound(a+1,a+1+n,r)-a-1;
        if(l<=r)
        {
            v1[l].push_back(r);
            v2[r].push_back(l);
        }
    }
    ll ans=1ll*n*(n-1)*(n-2)/6;
    for(int i=1; i<=n; ++i)
    {
        for(auto r: v1[i]) update(1,i,r,1,n);
        int res=0;
        if(i-1>=1) res+=i-1-query(1,1,i-1,1,n);
        if(i+1<=n) res+=query(1,i+1,n,1,n);
        ans-=1ll*res*(res-1)/2;
        for(auto l: v2[i]) update(1,l,i,1,n);
    }
    printf("%lld\n",ans);
    return 0;
}