Hdu 4893 Wow! Such Sequence! 線段樹單點更新,成段求和
阿新 • • 發佈:2019-02-12
題目大意:給你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; }