# 【洛谷5335】[THUSC2016] 补退选（指针实现Trie）

### 代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 100000
#define K 60
using namespace std;
class FastIO
{
private:
#define FS 100000
#define pc(c) (C==E&&(clear(),0),*C++=c)
#define tn (x<<3)+(x<<1)
#define D isdigit(c=tc())
int T;char c,*A,*B,*C,*E,FI[FS],FO[FS],S[FS];
public:
I FastIO() {A=B=FI,C=FO,E=FO+FS;}
Tp I void read(Ty& x) {x=0;W(!D);W(x=tn+(c&15),D);}
Tp I void write(Ty x) {x<0&&(pc('-'),x=-x);W(S[++T]=x%10+48,x/=10);W(T) pc(S[T--]);}
Tp I void writeln(Con Ty& x) {write(x),pc('\n');}
I void clear() {fwrite(FO,1,C-FO,stdout),C=FO;}
}F;
class Trie//指针实现Trie
{
private:
struct node
{
int V;node *S[10];vector<int> P;
I node() {V=0,P.clear();for(RI i=0;i^10;++i) S[i]=NULL;}//初始化清空节点，一定要写，不然会挂
}*rt;
public:
I Trie() {rt=new node;}//初始化根节点
I void Upt(Con string& s,CI v,CI ti)//修改（插入/删除）
{
node *x=rt;for(RI i=0,l=s.length(),t;i^l;++i)
x->S[t=s[i]-97]==NULL&&(x->S[t]=new node),x=x->S[t],//如果子节点为空，新建子节点
(x->V+=v)>x->P.size()&&(x->P.push_back(ti),0);//更新vector
}
I int Qry(Con string& s,CI v)//询问
{
node *x=rt;for(RI i=0,l=s.length(),t;i^l;++i)
if(x->S[t=s[i]-97]==NULL) return -1;else x=x->S[t];//如果子节点为空，直接返回-1
return x->P.size()>v?x->P[v]:-1;//判断是否有解，有解则返回答案，否则返回-1
}
}T;
int main()
{