1. 程式人生 > >Hdu 4893 Wow! Such Sequence! 線段樹單點更新,成段求和

Hdu 4893 Wow! Such Sequence! 線段樹單點更新,成段求和

題目大意:給你N個數,初始狀態每個數都為0,現在有三種操作:1,給第K個數增加a   2,詢問[L,R]區間所有數的和   3,區間【l,r】內每個數都變為離他最近的斐波那契數。

解題思路:剛開始想的時候  想用mark標記該區間是否已經變為離他最近的斐波那契數,每次插入一個數的時候進行向下向上更新,但是發現無法實時更新sum值。

正確的思路是:在每個節點開一個sum陣列記錄所有子節點的和,再開一個fibsum陣列記錄所有子節點的斐波那契數之和,然後進行操作3時,只需要更新值時將sum用fibsum更新就可以了,每次插入的時候要維護每個節點fibsum的值。程式碼如下:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<cstdlib>
using namespace std;
typedef __int64 LL;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define N 100005

LL sum[N<<4],fibsum[N<<4];
bool mark[N<<4];
LL fib[100];

void get_fib()
{
    fib[0] = fib[1] = 1;
    for(int i = 2; i <= 95; i++)
        fib[i] = fib[i-1] + fib[i-2];
}
LL nearest_fib(LL x)
{
	int pos = lower_bound(fib,fib+93,x)-fib;
	if(pos && abs(fib[pos-1]-x) <= abs(fib[pos]-x))
        return fib[pos-1];
	return fib[pos];
}
void pushup(int rt)//向上更新
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    fibsum[rt]=fibsum[rt<<1]+fibsum[rt<<1|1];
}

void pushdown(int rt)//向下更新
{
    if(mark[rt])
    {
        mark[rt<<1]=true;
        mark[rt<<1|1]=true;
        sum[rt<<1]=fibsum[rt<<1];
        sum[rt<<1|1]=fibsum[rt<<1|1];
        mark[rt]=false;
    }
}

void build(int l,int r,int rt)//建樹
{
    if(l==r)
    {
        mark[rt]=false;
        sum[rt]=0;
        fibsum[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
    mark[rt]=false;//建樹時所有節點標記為0
}
void insert(int pos,LL a,int l,int r,int rt)//單點插入值
{
    if(l==r&&l==pos)
    {
        sum[rt]+=a;
        fibsum[rt]=nearest_fib(sum[rt]);//更新fibsum【】的值
        return;
    }
    pushdown(rt);
    int m=(l+r)>>1;
    if(pos<=m) insert(pos,a,lson);
    else insert(pos,a,rson);
    pushup(rt);
}
void update(int L,int R,int l,int r,int rt)    //操作3區間更新最近值
{
    if(L<=l&&R>=r)
    {
        mark[rt]=true;
        sum[rt]=fibsum[rt];
        return;
    }
    pushdown(rt);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,lson);
    if(R>m)  update(L,R,rson);
    pushup(rt);
}

LL query(int L,int R,int l,int r,int rt)              //操作2區間求和
{
    if(L<=l&&R>=r)
        return sum[rt];
    pushdown(rt);
    int m=(l+r)>>1;
    LL ret=0;
    if(L<=m) ret+=query(L,R,lson);
    if(R>m)  ret+=query(L,R,rson);
    return ret;
}

int main()
{
    //freopen("in.txt","r",stdin);
    get_fib();
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        build(1,n,1);//建樹時已經初始化,故不需要memset,而且浪費時間
        int a,x,y;

        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&a,&x,&y);
            if(a==1)
                insert(x,y,1,n,1);
            else if(a==2)
                printf("%I64d\n",query(x,y,1,n,1));
            else
                update(x,y,1,n,1);
        }
    }
    return 0;
}