[Wannafly挑战赛14 B 前缀查询]（Trie+线段树）

题意：

在一个 Minecraft 村庄中，村长有这一本小写字母构成的名册（字符串的表），
每个名字旁边都记录着这位村民的声望值，而且有的村民还和别人同名。
随着时间的推移，因为没有村民死亡，这个名册变得十分大。
现在需要您来帮忙维护这个名册，支持下列 4 种操作：

1. 插入新人名 si，声望为 ai
2. 给定名字前缀 pi 的所有人的声望值变化 di
3. 查询名字为 sj 村民们的声望值的和（因为会有重名的）
4. 查询名字前缀为 pj 的声望值的和

题解：

插入字符串我们可以用Trie树存储，而对于2，3，4的操作，很容易想到线段树可以维护这些，最后再加个tag标记就好了。

我们用sum表示以当前字符串为前缀的声望总和

val表示特定字符串的声望总和

cnt表示以当前字符串为前缀的字符串数量

same表示特定字符串的数量

然后tag转移就好了

---

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN = 1e5+50;
struct node{
    int son[26];
    int cnt,same;
    LL sum,val,tag;
}seg[MAXN<<2];
char s[MAXN]; int tot=1;
inline void pushdown(int rt){
    if(rt==0 || seg[rt].tag==0) return;
    for(int i=0;i<26;i++){
        if(!seg[rt].son[i]) continue;
        seg[seg[rt].son[i]].tag += seg[rt].tag;
        seg[seg[rt].son[i]].sum += seg[seg[rt].son[i]].cnt*seg[rt].tag;
        seg[seg[rt].son[i]].val += seg[seg[rt].son[i]].same*seg[rt].tag;
    }
    seg[rt].tag=0;
}
inline void Insert(char *s,int val){
    int rt=0,len=strlen(s);
    for(int i=0;i<len;i++){
        pushdown(rt);
        if(!seg[rt].son[s[i]-'a']) seg[rt].son[s[i]-'a']=++tot;
        seg[rt].sum += val; seg[rt].cnt++;
        rt = seg[rt].son[s[i]-'a'];
    }
    pushdown(rt);
    seg[rt].sum += val; seg[rt].val += val;
    seg[rt].cnt++; seg[rt].same++;
}
inline void update(char *s,LL val){
    int rt=0,len=strlen(s);
    for(int i=0;i<len;i++){
        if(!seg[rt].son[s[i]-'a']) return;
        rt = seg[rt].son[s[i]-'a'];
    }
    LL cnt = seg[rt].cnt; rt=0;
    for(int i=0;i<len;i++){
        pushdown(rt);
        if(!seg[rt].son[s[i]-'a']) return;
        seg[rt].sum += cnt*val;
        rt = seg[rt].son[s[i]-'a'];
    }
    pushdown(rt);
    seg[rt].sum += seg[rt].cnt * val;
    seg[rt].val += seg[rt].same * val;
    seg[rt].tag += val;
}
LL Query(char *s,int tag){
    int rt=0,len=strlen(s);
    for(int i=0;i<len;i++){
        pushdown(rt);
        if(!seg[rt].son[s[i]-'a']) return 0;
        rt = seg[rt].son[s[i]-'a'];
    }
    pushdown(rt);
    if(tag==1) return seg[rt].sum;
    else return seg[rt].val;
}
int main(){
    //freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
    int n; scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int op; scanf("%d",&op);
        if(op==1){
            LL val; scanf("%s%lld",s,&val);
            Insert(s,val);
        }else if(op==2){
            LL val; scanf("%s%lld",s,&val);
            update(s,val);
        }else if(op==3){
            scanf("%s",s); printf("%lld\n",Query(s,2));
        }else{
            scanf("%s",s); printf("%lld\n",Query(s,1));
        }
    }
    return 0;
}