BNU 12680 Jupiter Atacks! (線段樹)

題意：給定一個區間，，兩種操作，一種是將x的值賦為y，另一種為詢問區間（x，y）的sum值，此處sum值的運算為：

例如： H(f2,..., f5) = B^0 * f5 + B^1 * f4 + B^2 * f3 + B^3 * f2 ( mod P)
這題隱藏的好深~~~ 用線段樹的話，用va域儲存樹內所有的H（tr[x].l，tr[x].r），更新時根據更新點和區間右端點乘以個（B ^ (tr[x].r - pos)）,求和時根據求和區間右端點和線段樹區間的右端點乘以個（B ^( r - tr[x].r））。
需要注意的是，取餘過程中可能會產生負數，需要歸正。

#include <iostream>#include <algorithm>#include <cmath>#include<functional>#include <cstdio>#include <cstdlib>#include <cstring>#include <string>#include <vector>#include <set>#include <queue>#include <stack>#include <climits>//形如INT_MAX一類的#define MAX 100050#define INF 0x7FFFFFFF#define L(x) x<<1#define R(x) x<<1|1using namespace std;int b,p,l,n;long long B[MAX],ans;int a[MAX];struct node {    int l,r,mid;    long long va;}tr[MAX * 4];void init() {    memset(a,0,sizeof(a));    B[0] = 1;    for(int i=1; i<=l; i++) {        B[i] = (B[i - 1] * b) % p;    }}void build(int l,int r,int x) {    tr[x].l = l;    tr[x].r = r;    tr[x].mid = (l + r) >> 1;    tr[x].va = 0;    if(l == r) return ;    build(l,tr[x].mid,L(x)) ;    build(tr[x].mid + 1, r, R(x));}void update(int l,int x,int va) {    tr[x].va = (tr[x].va + B[tr[x].r - l] * va % p ) % p;    if(tr[x].va < 0) tr[x].va += p;    if(tr[x].l == tr[x].r) return ;    if(l > tr[x].mid) update(l,R(x),va);    else update(l,L(x),va);}void query(int l,int r,int x) {    if(l <= tr[x].l && r >= tr[x].r) {        ans = (ans + B[r - tr[x].r] * tr[x].va % p) % p;        if(ans < 0) ans += p;        return ;    }    if(r > tr[x].mid) query(l,r,R(x));    if(l <= tr[x].mid) query(l,r,L(x));}void test() {    for(int i=0; i<=3 * l; i++) {        printf("%d %d %lld\n",tr[i].l,tr[i].r,tr[i].va);    }}int main() {    while(scanf("%d%d%d%d",&b,&p,&l,&n)) {        if(b == 0 && p == 0 && l == 0 && n == 0) break;        char op;        int x,y;        init();        build(1,l,1);        for(int i=0; i<n; i++) {            getchar();            scanf("%c%d%d", &op,&x,&y);            if(op == 'E') {                update(x,1,y - a[x]);                a[x] = y;            } else {                ans = 0;                //test();                query(x,y,1);                printf("%lld\n",ans);            }        }        puts("-");    }    return 0;}