這道題真的是好題,讓我對線段樹有了全新的認識,至少讓我真正感受到了線段樹的神奇。
題意是就是線段樹區間更新,單點詢問的問題,不過這個題好就好在它的區間更新的點并不連續!
adding c to each of Ai which satisfies a <= i <= b and (i - a) % k == 0.
關鍵在于更新的點所滿足的條件? 1.在區間[a,b]中。 2.關于k的余數與(a%k)相等。
其中第二個條件保證了更新的點不連續(但如果傳統的先判斷再更新,則線段樹失去了優勢,因為始終會超時)。
本題突破口在于K的大小(1 <= k <= 10),結合第二個條件,我們可以想到在線段樹的 Lazy標記上做文章。
當k==1 余數有 0
當k==2 余數有 0,1
當k==3 余數有 0,1,2
。。。。。
可以看到,只有55種情況,那么我們只需要將線段樹的 Lazy標記 擴充為大小55的數組,
然后根據相應的余數情況進行更新與查詢即可。
?
1 #include<iostream> 2 #include<sstream> 3 #include<stack> 4 #include<cstdio> 5 #include<cstdlib> 6 #include<cstring> 7 #include<climits> 8 #include<cctype> 9 #include<queue> 10 #include<algorithm> 11 #include<cmath> 12 #include<map> 13 #include<set> 14 #define lson root<<1,l,mid 15 #define rson root<<1|1,mid+1,r 16 17 #define inf 0x3f3f3f3f 18 #define N 50010 19 #define maxn 10001000 20 #define mod 1000000007 21 using namespace std; 22 23 int a[N]; 24 int b[11][11]; //用于將55種余數情況一 一 對 應 25 struct NODE{ 26 int l,r; 27 int v; 28 int b[55]; 29 int mid(){ 30 return (l+r)>>1; 31 } 32 }node[N<<2]; 33 34 void build(int root,int l,int r) 35 { 36 node[root].l=l; 37 node[root].r=r; 38 node[root].v=0; 39 memset(node[root].b,0,sizeof(node[root].b)); 40 if(l==r) return; 41 int mid=node[root].mid(); 42 build(lson); 43 build(rson); 44 } 45 46 void pushdown(int root) //關于pushdown函數,個人覺得只需要詢問時調用就行了,而且也過了,應該沒問題,有的話請指出,謝謝 47 { 48 for(int i=1; i<=10; ++i) 49 { 50 for(int j=0; j<i; ++j) 51 { 52 node[root<<1].b[b[i][j]]+=node[root].b[b[i][j]]; 53 node[root<<1|1].b[b[i][j]]+=node[root].b[b[i][j]]; 54 } 55 } 56 memset(node[root].b,0,sizeof(node[root].b)); 57 } 58 59 void update(int x,int y,int root,int v,int k,int rd) 60 { 61 if(node[root].l==x&&node[root].r==y) 62 { 63 node[root].b[b[k][rd]]+=v; 64 node[root].v+=v; 65 return; 66 } 67 int mid=node[root].mid(); 68 if(y<=mid) update(x,y,root<<1,v,k,rd); 69 else if(x>mid) update(x,y,root<<1|1,v,k,rd); 70 else 71 { 72 update(x,mid,root<<1,v,k,rd); 73 update(mid+1,y,root<<1|1,v,k,rd); 74 } 75 } 76 77 int query(int root,int x) 78 { 79 if(node[root].l==x&&node[root].r==x) 80 { 81 int ans=0; 82 for(int i=1; i<=10; ++i) 83 ans+=node[root].b[b[i][x%i]]; 84 return ans; 85 } 86 pushdown(root); 87 int mid=node[root].mid(); 88 if(x<=mid) return query(root<<1,x); 89 else return query(root<<1|1,x); 90 } 91 92 int main() 93 { 94 // freopen("lxx.txt","r",stdin); 95 int group,figure,i,j,x,y,k,v,num; 96 int cnt=0; 97 for(i=1; i<=11; ++i) 98 for(j=0; j<i; ++j) 99 b[i][j]=cnt++; 100 while(~scanf("%d",&group)) 101 { 102 build(1,1,group); 103 memset(a,0,sizeof(a)); 104 for(i=1; i<=group; ++i) scanf("%d",&a[i]); 105 scanf("%d",&group); 106 while(group--) 107 { 108 scanf("%d",&num); 109 if(num==1) 110 { 111 scanf("%d%d%d%d",&x,&y,&k,&v); 112 update(x,y,1,v,k,x%k); 113 } 114 else 115 { 116 scanf("%d",&figure); 117 printf("%d\n",query(1,figure)+a[figure]); 118 } 119 } 120 } 121 return 0; 122 }
?
不過感覺自己的代碼風格還不夠成熟,繼續努力!
?