https://codeforces.com/gym/102465/problem/G
没打cf又摸鱼到这么晚。。。心虚下床看看队友写的题。。。下午想了好久没想出怎么做
这题转换成求前缀和以后就很好做了,如果是拼接的话,那么直接把权值和长度加起来就行,如果是取一段,我们考虑递归的去求lo和hi的前缀和然后相减就行
然后递归的时候,我们就是求s[id]串的前x的前缀和
如果s[id]是拼接的,那么如果x<=len[a]就直接a串里面递归,否则就加上val[a]整个串的值,再去递归b串,加上b串部分的值。
如果s[id]也是截取的,那么就去递归s[id]截取的那个串s[s[id]]求右端点的前缀和,然后由于s[id]在截取s[s[id]]过程中的左区间的前缀和pre[s[id]]之前已经求过了,那么就是右端点前缀和减去左端点前缀和就行了
#include <bits/stdc++.h> using namespace std; typedef long long ll; const ll N=3000,mo=1e9+7; ll sum[N]; ll val[N],pre[N],suf[N],len[N]; ll A[N],B[N],C[N],D[N]; char s[N]; ll findpre(int id,ll x){ //qian x if(id==0){ if(x==0) return 0; return sum[x-1]; } if(x==len[id]) return val[id]; if(A[id]==2){ int a=B[id],b=C[id]; if(x<=len[a]) return findpre(a,x); else return (val[a]+findpre(b,x-len[a]))%mo; } else{ long long ans=findpre(D[id],B[id]+x); ans-=pre[id]; return ans%mo; } } int main(){ int i,n,a; scanf("%d",&n); scanf("%s",s); len[0]=strlen(s); for(i=0;s[i];i++){ if(i==0) sum[i]=s[i]; else sum[i]=sum[i-1]+s[i]; val[0]+=s[i]; } for(i=1;i<n;i++){ scanf("%s",&s); if(s[0]=='S'){ scanf("%lld%lld%lld",&D[i],&B[i],&C[i]); //C[i]--; A[i]=1; pre[i]=findpre(D[i],B[i]); suf[i]=findpre(D[i],C[i]); val[i]=(suf[i]-pre[i])%mo; len[i]=C[i]-B[i]; } else{ scanf("%lld%lld",&B[i],&C[i]); A[i]=2; val[i]=(val[B[i]]+val[C[i]])%mo; len[i]=len[B[i]]+len[C[i]]; } val[i]%=mo; if(val[i]<0) val[i]+=mo; } // cout<<pre[1]<<" "<<suf[1]<<endl; cout<<val[n-1]<<endl; }