LOJ 2991 「THUSC 2016」补退选——trie+线段树合并或vector

题目：https://loj.ac/problem/2291

想了线段树合并的做法。就是用线段树维护 trie 的每个点在各种时间的操作。

然后线段树合并一番，线段树维护前缀最大值，就是维护最大子段和的套路，记录区间和、前缀 max 。查询的时候，因为当前区间只记录了自己区间内部的前缀 max 值，所以要加一个 pr 表示该区间前面的区间和。

空间可能爆？ RE 就没管。后来发现是 go[ ][ ] 开成 N 而非 M 了。这个做法还是可过的。

注意强制在线的 ans 是带绝对值的。注意 mx 的初值是 0 而非 -INF 因为可以不选。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
int Mx(int a,int b){return a>b?a:b;}
int Mn(int a,int b){return a<b?a:b;}
const int N=1e5+,M=6e6+,M2=2e7+;//mention
int n,cnt=,go[M][],rt[M],qnt;//M not N!!!
int tot,Ls[M2],Rs[M2],sm[M2],mx[M2]; char ch[];
struct Node{ int a,b,c,t;char s[];}q[N];
int nwnd(int pr)
{
  int cr=++tot; ls=Ls[pr]; rs=Rs[pr];
  sm[cr]=sm[pr]; mx[cr]=mx[pr]; return cr;
}
void pshp(int cr)//ini mx is 0!!(can choosen't)
{
  mx[cr]=Mx(mx[ls],sm[ls]+mx[rs]);
  sm[cr]=sm[ls]+sm[rs];
}
void ins(int l,int r,int &cr,int p,int k)
{
  if(!cr)cr=++tot;
  if(l==r){sm[cr]=mx[cr]=k;return;}
  int mid=l+r>>;
  if(p<=mid)ins(l,mid,ls,p,k);
  else ins(mid+,r,rs,p,k);
  pshp(cr);
}
void ins(int t,int k)
{
  int cr=,len=strlen(ch+);
  for(int i=,d;i<=len;i++)
    {
      d=ch[i]-'a';
      if(!go[cr][d])go[cr][d]=++cnt;
      cr=go[cr][d];
    }
  ins(,n,rt[cr],t,k);
}
void mrg(int l,int r,int &cr,int pr)
{
  if(!pr)return; if(!cr){cr=pr;return;}
  cr=nwnd(cr); int mid=l+r>>;
  mrg(l,mid,ls,Ls[pr]); mrg(mid+,r,rs,Rs[pr]);
  pshp(cr);
}
void dfs(int cr)
{
  for(int i=,v;i<;i++)
    if(v=go[cr][i])
      {
    dfs(v);
    mrg(,n,rt[cr],rt[v]);
      }
}
int qry(int l,int r,int cr,int R,int lm,int pr)
{
  if(pr+mx[cr]<=lm)return -; if(l==r)return l;
  int mid=l+r>>;
  if(mid>=R)return qry(l,mid,ls,R,lm,pr);
  if(pr+mx[ls]>lm)return qry(l,mid,ls,R,lm,pr);
  return qry(mid+,r,rs,R,lm,pr+sm[ls]);
}
int qry(int bh,int lm)
{
  int cr=,len=strlen(q[bh].s+);
  for(int i=,d;i<=len;i++)
    {
      d=q[bh].s[i]-'a';
      if(!go[cr][d])return -;//-1 not 0!!!!!
      cr=go[cr][d];
    }
  return qry(,n,rt[cr],q[bh].t,lm,);
}
int main()
{
  n=rdn();
  for(int i=,op;i<=n;i++)
    {
      op=rdn();
      if(op==||op==)
    {
      scanf("%s",ch+);
      ins(i,(op==?:-));
    }
      else
    {
      qnt++; scanf("%s",q[qnt].s+);
      q[qnt].a=rdn(); q[qnt].b=rdn();
      q[qnt].c=rdn(); q[qnt].t=i;
    }
    }
  dfs(); int ans=;
  for(int i=;i<=qnt;i++)
    {
      if(ans<)ans=-ans;//////
      int k=((long long)q[i].a*ans+q[i].b)%q[i].c;
      ans=qry(i,k);
      printf("%d\n",ans);
    }
  return ;
}

看了题解，原来 trie 每个节点开 vector 存 “超过某个值的最早时间” 即可。也不用往上合并，往下走的时候路径上的 vector 全加上即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define pb push_back
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
const int N=1e5+,M=N*;
int n,go[M][],tot=,ct[M],ans; char ch[];
vector<int> vt[M];
void ins(int t,int k)
{
  int cr=,len=strlen(ch+);
  for(int i=,d;i<=len;i++)
    {
      d=ch[i]-'a';
      if(!go[cr][d])go[cr][d]=++tot;
      cr=go[cr][d]; ct[cr]+=k;
      if(ct[cr]>vt[cr].size())
    vt[cr].pb(t);
    }
}
void qry(int k)
{
  int cr=,len=strlen(ch+);
  for(int i=,d;i<=len;i++)
    {
      d=ch[i]-'a';
      if(!go[cr][d]){ans=-;printf("%d\n",ans);return;}
      cr=go[cr][d];
    }
  if(vt[cr].size()<=k)ans=-;
  else ans=vt[cr][k];
  printf("%d\n",ans);
}
int main()
{
  n=rdn();
  for(int i=,op;i<=n;i++)
    {
      op=rdn();scanf("%s",ch+);
      if(op==||op==)ins(i,(op==?:-));
      else
    {
      int a=rdn(),b=rdn(),c=rdn();
      if(ans<)ans=-ans;
      int k=((long long)a*ans+b)%c;
      qry(k);
    }
    }
  return ;
}