hihoCoder 1116 计算（线段树）

http://hihocoder.com/problemset/problem/1116

题意：

思路：

用线段树解决，每个节点需要设置4个变量，sum记录答案，all记录整个区间的乘积，pre记录该区间内所有前缀的和，suf记录该区间内所有后缀的和。

举个例子，比如现在要合并{a,b,c,d}和{x,y,z,w}这两个区间，那么新增的答案有哪些呢？

有abcdx（省去乘号)，abcdxy,abcdxyz,abcdxyzw，bcdxyzw，cdxyzw，dxyzw。这些就是{a,b,c,d}的所有后缀×{x,y,z,w}所有前缀。

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 100000+5;
const int mod = 10007;
typedef long long ll;

int n,q;

struct node
{
    int l,r;
    ll sum, pre, suf, all;
}t[maxn<<4];


void PushUp(int o)
{
    t[o].all=(t[o<<1].all*t[o<<1|1].all)%mod;
    t[o].sum=((t[o<<1].sum+t[o<<1|1].sum)%mod+(t[o<<1].suf*t[o<<1|1].pre)%mod)%mod;
    t[o].pre=(t[o<<1].pre+(t[o<<1].all*t[o<<1|1].pre)%mod)%mod;
    t[o].suf=(t[o<<1|1].suf+(t[o<<1|1].all*t[o<<1].suf)%mod)%mod;
}

void build(int l, int r, int o)
{
    t[o].l = l;
    t[o].r = r;
    t[o].sum = t[o].pre = t[o].suf = t[o].all = 0;
    if(l==r)  return;
    int mid = (l+r)>>1;
    build(l,mid,o<<1);
    build(mid+1,r,o<<1|1);
    PushUp(o);
}

void update(int l, int r, int x, int v, int o)
{
    if(t[o].l == x && t[o].l==t[o].r)
    {
        t[o].sum = t[o].pre = t[o].suf = t[o].all = v;
        return;
    }
    int mid = (l+r)>>1;
    if(x<=mid)  update(l,mid,x,v,o<<1);
    else update(mid+1,r,x,v,o<<1|1);
    PushUp(o);
}

int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&q);
    build(1,n,1);
    while(q--)
    {
        ll x,y;
        scanf("%lld%lld",&x,&y);
        update(1,n,x,y%mod,1);
        printf("%lld
",t[1].sum);
    }
    return 0;
}