题意:
在一个 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;
}