Codeforces Round #307 (Div. 2) 551E - GukiZ and GukiZiana 分块

来源：互联网发布：经济全球化的数据编辑：程序博客网时间：2024/05/08 14:28
题意:就是给一个数列，有两种操作
1 l r x从l~r都+x
2 x查询最右边==x的位置-最左边==x的位置
思路:
分块一下就行了
#include<stdio.h>#include<string.h>#include<iostream>#include<algorithm>#include<math.h>#include<queue>#include<stack>#include<string>#include<vector>#include<map>#include<set>using namespace std;#define mem(a,b) memset(a,b,sizeof(a))#define lowbit(x) (x&(-x))typedef long long LL;#define maxn 550005const int inf=10000005;LL A[maxn],pos[maxn],add[maxn];struct node{    LL val;    int pos;    node(LL a=0,int b=0)    {        val=a,pos=b;    }    bool friend operator <(node u,node v)    {        return u.val<v.val;    }};vector<node>vec[50005];void Add(int l,int r,LL x){    int left=pos[l],right=pos[r];    for(int i=left+1;i<right;++i)    add[i]+=x;    if(left==right)    {        int Size=vec[left].size();        for(int i=0;i<Size;++i)        if(vec[left][i].pos>=l&&vec[left][i].pos<=r)        vec[left][i].val+=x;        sort(vec[left].begin(),vec[left].end());    }    else    {        int Size=vec[left].size();        for(int i=0;i<Size;++i)        if(vec[left][i].pos>=l&&vec[left][i].pos<=r)        vec[left][i].val+=x;        Size=vec[right].size();        for(int i=0;i<Size;++i)        if(vec[right][i].pos>=l&&vec[right][i].pos<=r)        vec[right][i].val+=x;        sort(vec[left].begin(),vec[left].end());        sort(vec[right].begin(),vec[right].end());    }}int query(int pos,LL x){    x-=add[pos];    int r=vec[pos].size()-1,l=0;    while(l<=r)    {        int mid=(l+r)/2;        if(vec[pos][mid].val==x) return 1;        if(vec[pos][mid].val>x) r=mid-1;        else l=mid+1;    }    return 0;}int find_min(int pos,LL x){    x-=add[pos];    int Size=vec[pos].size();    int tmp=inf;    for(int i=0;i<Size;++i)    if(vec[pos][i].val==x)    tmp=min(tmp,vec[pos][i].pos);    return tmp;}int find_max(int pos,LL x){    x-=add[pos];    int Size=vec[pos].size();    int tmp=-inf;    for(int i=0;i<Size;++i)    if(vec[pos][i].val==x)    tmp=max(tmp,vec[pos][i].pos);    return tmp;}int main(){    int n,q;    scanf("%d%d",&n,&q);    for(int i=1;i<=n;++i)    scanf("%lld",&A[i]);    int t=sqrt(n),pos1=t,pos2=1;    for(int i=1;i<=n;++i)    if(i<=pos1)    {        vec[pos2].push_back(node(A[i],i));        pos[i]=pos2;    }    else    {        pos1+=t;pos2++;        vec[pos2].push_back(node(A[i],i));        pos[i]=pos2;    }    for(int i=1;i<=pos2;++i)    sort(vec[i].begin(),vec[i].end());    int tot=0;    while(q--)    {        LL x;        scanf("%lld",&x);        if(x==1)        {            int l,r;            scanf("%d%d%lld",&l,&r,&x);            Add(l,r,x);        }        else        {            tot++;            scanf("%lld",&x);            int l=inf,r=-inf;            for(int i=1;i<=pos2;++i)            if(query(i,x))            {                //printf("%d %d\n",l,r);                l=min(i,l);r=max(i,r);                //printf("***%d %d %d\n",l,r,i);            }            if(l!=inf)            {                l=find_min(l,x);r=find_max(r,x);                printf("%d\n",r-l);            }            else printf("-1\n");            //printf("***%d %d\n",l,r);        }        /*printf("%d\n",pos2);        for(int i=1;i<=pos2;++i)        {            int Size=vec[i].size();            for(int j=0;j<Size;++j)            printf("%lld(%d) ",vec[i][j].val+add[i],vec[i][j].pos);            printf("\n");        }*/    }    return 0;}
0 0