2020算法设计竞赛 H 坐火车

链接：https://ac.nowcoder.com/acm/contest/3005/H
来源：牛客网

大致题意：让我们针对每一个数，求这个数左区间和右区间颜色相同（也就是数字相同）得对数；

比如：左边3个“3'得颜色，右边2个‘3’得颜色，就有2*3=6对；

数据范围为5e5；所以可接受得复杂度为nlogn；

那么我们可以考虑线段树；

如何维护呢？

我们用两个vis去标记左边和右边各种颜色的个数；代码中visa标记的是右区间，visb标记的是左区间；

然后从左往右遍历一遍：

　　在某一次操作中，我们对第K个数求值；那么我们需要先把右区间颜色k的减1；

这个时候假如左区间有这个K数，我们就需要将这个数包含的对数求出来，然后update，

这里有个细节，假如k-1==k,那么需要减掉的对数要减1；

因为k-1并没有在上次的操作中计算；

然后，我们就求K-1这个位置的颜色在右区间的个数，然后将这个数update；

在跑一遍线段树求出第K个数的query（l,r,1）即可；

 

 

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int maxn=5e5+;
 int visa[maxn];
 int visb[maxn];
 ll ans[maxn];
 struct node
 {
     int val,l,r;
 }a[maxn];
 struct haa
 {
     int l;int r;
     ll sum;
 }tree[maxn<<];
 void build(int l,int r,int root)
 {
     tree[root].l=l;tree[root].r=r;
     tree[root].sum=;
     if(l==r) return;
     int mid=l+r>>;
     build(l,mid,root<<);
     build(mid+,r,root<<|);
 }
 void up(int root)
 {
     tree[root].sum=tree[root<<].sum+tree[root<<|].sum;
 }
 void update(int pos,int val,int root)
 {
     int L=tree[root].l;
     int R=tree[root].r;
     if(L==R){
         tree[root].sum+=val;
         return;
     }
     int mid=L+R>>;
     if(pos<=mid) update(pos,val,root<<);
     else update(pos,val,root<<|);
     up(root);
 }
 ll query(int l,int r,int root)
 {
     int L=tree[root].l;
     int R=tree[root].r;
     if(l<=L&&r>=R){
         return tree[root].sum;
     }
     ll ans=;
     int mid=L+R>>;
     if(l<=mid) ans+=query(l,r,root<<);
     if(r>mid)  ans+=query(l,r,root<<|);
     return ans;
 }
 int main()
 {
     int n;
     scanf("%d",&n);
     for(int i=;i<=n;i++){
         scanf("%d%d%d",&a[i].val,&a[i].l,&a[i].r);
         visa[a[i].val]++;
     }
     build(,maxn,);
     ans[]=;
     visa[a[].val]--;  //右区间--
     visb[a[].val]++;  //左区间++；
     for(int i=;i<=n;i++){
         visa[a[i].val]--; //右区间--；
         if(visb[a[i].val]){ //如果左区间包含这个数的话；
             //我们就需要减掉与这个数相关的对数；
             //但是左区间包含i-1这一项；而这一项在上次操作中没有出现；
             //所以当两者相等的时候，tmp--;
             int tmp=visb[a[i].val];
             if(a[i-].val==a[i].val) tmp--;
             update(a[i].val,-tmp,);
         }
         int tmp=visa[a[i-].val];
         update(a[i-].val,tmp,);
         ans[i]=query(a[i].l,a[i].r,);
         visb[a[i].val]++;
     }
     for(int i=;i<=n;i++)
         printf("%lld ",ans[i]);
     printf("\n");
     return ;
 }