GZOI 2017配对统计 树状数组

题目

https://www.luogu.com.cn/problem/P5677

分析

　　最开始读题的时候没有读的太懂，以为i是在选定区间内给的，实际上不是，这道题的意思应该是在l和r的区间内找出有多少个好的配对，这里好的配对是对于整个区间来说的，既然是对于整个区间，我们就不难想到找出好的配对的方法，所以我们可以先找出所有好的配对，然后用树状数组维护个数。

　　如何找出好的配对呢？我们先来分析什么叫好的配对，选定的两点间距离比其中一点到除对方外任意一点的距离都小，也就是说这两点差的绝对值最小，这样的话，这两个点在sort排序后一定相邻，这个很好推出，于是我们只要考虑这个点的另一个配对是左边的点还是右边的点，写一个判断就行了，注意特判1和n。

　　找出好的配对来，就又向答案接近了一步，现在我们只要进行更新就行了，这里的更新我们枚举左端点，上一步我们已经求出了好的配对（l，r）如果查询的左端点在l的左边，那么从r开始向右就一定至少存在一个好的配对，所以让树状数组中的r对应的位置更新就行。注意我们要倒序枚举左端点，因为我们加入r后所产生的配对只能是在询问区间包含l的情况下才有效。

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 using namespace std;
 const int N=3e5+;
 #define ll long long
 struct Node{
     ll val,id;
     bool operator < (const Node&A)const{
         return val<A.val;
     }
 }a[N];
 vector<int> point[N],p1[N],p2[N];
 ll lowbit(ll x){
     return x&-x;
 }
 void Ins(ll x,ll y){
     point[min(x,y)].push_back(max(x,y));//min表示左端点，max表示右端点
 }
 ll m,n,c[N];
 void Add(ll x){
     while(x<=n){
         c[x]++;
         x+=lowbit(x);
     }
 }
 ll query(ll x){
     ll ans=;
     while(x){
         ans+=c[x];
         x-=lowbit(x);
     }
     return ans;
 }
 ll ans[N];
 int main(){
     scanf("%lld%lld",&n,&m);
     for(int i=;i<=n;i++){
         scanf("%lld",&a[i].val);a[i].id=i;
     }
     sort(a+,a+n+);
     for(int i=;i<=n;i++){
         int Min=0x3f3f3f3f;
         if(i!=&&a[i].val-a[i-].val<Min){
             Min=a[i].val-a[i-].val;
         }
         if(i!=n&&a[i+].val-a[i].val<Min){
             Min=a[i+].val-a[i].val;
         }
         if(i!=&&a[i].val-a[i-].val==Min)Ins(a[i].id,a[i-].id);
         if(i!=n&&a[i+].val-a[i].val==Min)Ins(a[i+].id,a[i].id);//取两个端点最小值
     }
     for(int i=;i<=m;i++){
         int l,r;
         scanf("%d%d",&l,&r);
         p1[l].push_back(r);
         p2[l].push_back(i);//记录坐标位置
     }
     for(int i=n;i>=;i--){
         for(int j=;j<(int)point[i].size();j++)Add(point[i][j]);//倒序查询
         for(int j=;j<(int)p1[i].size();j++)ans[p2[i][j]]=query(p1[i][j]);
     }
     ll sum=;
     for(int i=;i<=m;i++)
         sum+=ans[i]*i;
     printf("%lld\n",sum);
 }