hdu 3074 Multiply game

线段树简单题：区间乘积，单点修改

这题其实难度和之前的一样本来不想做的，但是学习了高手的代码，所以自己想实现一遍

果然用了新的代码风格，时间大大提高，空间都减少，不过代码量好像没怎么变，一口气冲进了第10名

10

G-rated

281MS

1280K

1572B

C++

2013-01-31 12:57:58

#include <cstdio>

#define N 50010

const int MOD=1000000007;

__int64 mul[4*N];

int n,m;



void updata(int a ,int b ,int p ,int e ,int root)

{

    if(a==b)

    {

        mul[root]=e;

        return ;

    }

    int mid=(a+b)>>1;

    if(p<=mid)

        updata(a,mid,p,e,root<<1);

    else

        updata(mid+1,b,p,e,root<<1|1);

    mul[root]=(mul[root<<1]*mul[root<<1|1])%MOD;

    return ;

}



__int64 query(int a ,int b ,int l ,int r ,int root)

{

    int mid=(a+b)>>1;

    if(a==l && b==r)

        return mul[root];

    if(r<=mid)

        return query(a,mid,l,r,root<<1);

    else if(l>mid)

        return query(mid+1,b,l,r,root<<1|1);

    else

        return (query(a,mid,l,mid,root<<1)*query(mid+1,b,mid+1,r,root<<1|1))%MOD;

}



void build(int a ,int b , int root)

{

    if(a==b)

    {

        scanf("%I64d",&mul[root]);

        //printf("[%d , %d] , %d\n",a,b,mul[root]);

        return ;

    }



    int mid=(a+b)>>1;

    build(a,mid,root<<1);

    build(mid+1,b,root<<1 | 1);

    mul[root]=(mul[root<<1]*mul[root<<1 | 1])%MOD;

    //printf("[%d , %d] , %d\n",a,b,mul[root]);

    return ;

}



int main()

{

    int T;

    scanf("%d",&T);

    while(T--)

    {

        scanf("%d",&n);

        build(1,n,1);

        scanf("%d",&m);

        while(m--)

        {

            int k,a,b;

            scanf("%d%d%d",&k,&a,&b);

            if(!k)  //查询区间乘积

                printf("%I64d\n",query(1,n,a,b,1));

            else    //更改单点数值

                updata(1,n,a,b,1);

        }

    }

    return 0;

}

 

 

关于query函数的另一个写法

实际上我更喜欢上面的写法，因为更容易理解，符合人的思维。而且从时间上看，上面的写法比接下来的这个写法要更快。

不过这个写法代码量就少一些

这个写法是基于线段树的性质：线段树中任意两个节点的关系要么是包含关系，要么是相离关系，不可能出现相交

 

__int64 query(int a ,int b ,int l ,int r ,int root)

{

    if(r<a || l>b)  return 1; //完全无交集

    if(l<=a && b<=r) return mul[root];

    int mid=(a+b)>>1;

    return (query(a,mid,l,r,root<<1)*query(mid+1,b,l,r,root<<1|1))%MOD;

}