codeforces 292E Copying Data（線段樹）

http://codeforces.com/problemset/problem/292/E

題目大意：兩個數組，a和b，兩種操作：

1：將數組a的區間[x,x+k-1]複製給數組b的區間[y,y+k-1]。

2：問當前b[x]的值。

 

思路：這道題可以用線段樹做，線上段樹中維護以下值：sta，stb，這兩個值只有線上段樹的葉子節點才有意義，對於一個位置po，（也就是線段[po,po]，它是線段樹的葉子節點），sta為0表示這個位置還沒有被a覆蓋，保持原來的值，否則，sta表示位置po已被覆蓋，且是被a[sta]~a[sta+k-1]覆蓋。stb為0和sta類似，不為0表示該位置是從stb開始被覆蓋的。下面舉個例子。

比如我們進行操作1，將a[x]~a[x+k-1]的值複製給數組b的區間[y,y+k-1],那麼我們將線段樹中區間[y,y+k-1]中的sta賦值為x,將stb賦值為y。

那麼對於操作1,：1,x,y,k，我們就將區間[y,y+k-1]中的sta，stb分別賦值為x,y.

對於操作2:2,po 我們得到位置po的值為sta和stb，若sta為0，則輸出b[po]即可，否則輸出a[po-stb+sta]即可。

代碼如下：

#include <iostream>#include <string.h>#include <algorithm>#include <stdio.h>#define maxn 100010#define mid ((t[p].l+t[p].r)>>1)#define ls (p<<1)#define rs (ls|1)using namespace std;struct tree{    int l,r;    int sta,stb;}t[maxn<<2];int a[maxn],b[maxn];void pushdown(int p){    if(t[p].sta)    {        t[ls].sta=t[p].sta;        t[ls].stb=t[p].stb;        t[rs].sta=t[p].sta;        t[rs].stb=t[p].stb;        t[p].sta=0;    }}void build(int p,int l,int r){    t[p].l=l,t[p].r=r,t[p].sta=t[p].stb=0;    if(l==r)    return;    build(ls,l,mid);    build(rs,mid+1,r);}void change(int p,int l,int r,int sta,int stb){    if(t[p].l==l&&t[p].r==r)    {        t[p].sta=sta;        t[p].stb=stb;        return;    }    pushdown(p);    if(r<=mid)    change(ls,l,r,sta,stb);    else if(l>mid)    change(rs,l,r,sta,stb);    else    {        change(ls,l,mid,sta,stb);        change(rs,mid+1,r,sta,stb);    }}struct node{    int sta,stb;};node query(int p,int x){    node tmp;    if(t[p].l==t[p].r)    {        tmp.sta=t[p].sta,tmp.stb=t[p].stb;        return tmp;    }    pushdown(p);    if(x>mid)    return query(rs,x);    return query(ls,x);}int main(){    freopen("dd.txt","r",stdin);    int n,q,i;    scanf("%d%d",&n,&q);    for(i=1;i<=n;i++)    scanf("%d",&a[i]);    for(i=1;i<=n;i++)    scanf("%d",&b[i]);    build(1,1,n);    while(q--)    {        int t,x,y,z;        scanf("%d",&t);        if(t==1)        {            scanf("%d%d%d",&x,&y,&z);            change(1,y,y+z-1,x,y);        }        else        {            scanf("%d",&x);            node tmp=query(1,x);            if(tmp.stb==0)            printf("%d\n",b[x]);            else            printf("%d\n",a[x-tmp.stb+tmp.sta]);        }    }    return 0;}