CodeForces - 551E GukiZ and GukiZiana （分块 + 二分搜索）

题目链接：http://codeforces.com/problemset/problem/551/E

题目大意：

定义两种操作，一种区间增加一个值

另一种查找整个数组中一个指定数，求出最右边这个数的位置-最左边这个数的位置，如果没有就为-1

解题思路：

既要二分又要局部修改导致排序后的块内顺序错乱。

所以用vector储存一个块内排序后的数组。

然后对于区间修改时边界两个不完整的块暴力处理，处理完后重新调整对应vector数组

每次询问，正向搜一遍块，二分找出最左的包含那个数字的块，然后在原数组中暴力寻找真正的位置（二分找到的是在这个块中排序后的位置）

然后反向搜一遍，找出最右的。

 

代码中，atag[]是区间整体增加的值，每个点实际值=atag[对应块]+该点本身的值

原数组感觉从v[1]开始输入比较好。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#define ll long long
#define debug(x) printf("----Line:%s----\n",#x)
#define pb push_back
using namespace std;

const int N = 5e5+5;
const int M = sqrt(N+0.5)+5;

vector<ll>ve[M];
ll atag[M],v[N];
int blo,bl[N],n;

void reset(int k)
{
    ve[k].clear();
    for (int i=blo*(k-1)+1;i<=min(blo*k,n);i++)
        ve[k].pb(v[i]);
    sort(ve[k].begin(),ve[k].end());
}

void update(int l,int r,ll x)
{
    for (int i=l;i<=min(r,blo*bl[l]);i++)
        v[i] += x;
    reset(bl[l]);
    if (bl[l]!=bl[r]){
        for (int i=blo*(bl[r]-1)+1;i<=r;i++)
            v[i] += x;
        reset(bl[r]);
    }
    for (int i=bl[l]+1;i<=bl[r]-1;i++)
        atag[i] += x;
}

int query(ll x)
{
    int ansl,ansr;
    int pos = -1;
    for (int i=1;i<=bl[n];i++)
        if ( binary_search(ve[i].begin(),ve[i].end(),x-atag[i]) ){
            pos = i;
            break;
        }

    if (pos==-1) return -1;

    for (int i=blo*(pos-1)+1;i<=min(blo*pos,n);i++)
        if (v[i]+atag[pos]==x) {
            ansl = i;
            break;
        }
    for (int i=bl[n];i>=1;i--)
        if ( binary_search(ve[i].begin(),ve[i].end(),x-atag[i]) ) {
            pos = i;
            break;
        }
    for (int i=min(pos*blo,n);i>=blo*(pos-1)+1;i--)
        if (v[i]+atag[pos]==x) {
            ansr = i;
            break;
        }

    return ansr-ansl;
}

int main()
{
    int t,l,r,op;
    ll x;
    while (~scanf("%d %d",&n,&t)){
        blo = sqrt(n+0.5);
        for (int i=1;i<=M;i++) ve[i].clear();
        memset(atag,0,sizeof atag);
        for (int i=1;i<=n;i++){
            scanf("%I64d",v+i);
            bl[i] = (i-1)/blo+1;
            ve[bl[i]].pb(v[i]);
        }
        for (int i=1;i<=bl[n];i++) sort(ve[i].begin(),ve[i].end());
        while (t--){
            scanf("%d",&op);
            if (op==1) scanf("%d %d %I64d",&l,&r,&x),update(l,r,x);
            else scanf("%I64d",&x),printf("%d\n",query(x));
        }
    }
}

第三次做题了，边界还是会搞错，真实蒻鷄，这个礼拜好好学一下分块。