超级记事本([IOI2012]龙虾抄写器-字典树)



我记得以前写过一道题和这个一样。。。

时隔一年,,自己的Trie树究竟长进多少?


#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#include<cmath>
#include<cctype>
#include<cassert>
#include<climits>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define RepD(i,n) for(int i=n;i>=0;i--)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define INF (2139062143)
#define F (1000000009)
#define MAXN (100000+10)
#define LogMAXN (20+10)
typedef long long ll;
struct Tnode
{
   char c;
   int fa[LogMAXN],deep;
   Tnode():deep(0){MEM(fa) }
   Tnode(char _c,int _fa,int _deep):c(_c),deep(_deep){MEM(fa) fa[0]=_fa;}
}a[MAXN];
int n,tot=1;
void msm(int t)
{
   for(int i=1;a[t].fa[i-1];i++)
      a[t].fa[i]=a[a[t].fa[i-1]].fa[i-1];
}
void type(char c)
{
   tot++;
   a[tot]=Tnode(c,tot-1,a[tot-1].deep+1);msm(tot);
}
void query(int p)
{
   p=a[tot].deep-p;
   int now=tot;
   while(p)
   {
      int t=(int)(trunc(log2(p)));
      now=a[now].fa[t];
      p-=1<<t;
   }
   printf("%c\n",a[now].c);
}
void undo(int p)
{
   tot++;
   a[tot]=a[tot-1-p];
}
int main()
{
   freopen("spnp.in","r",stdin);
   freopen("spnp.out","w",stdout);
   scanf("%d",&n);
   For(i,n)
   {
      char s[2],c;int p;
      scanf("%s",s);
      switch(s[0])
      {
         case'T':scanf(" %c",&c);type(c);break;
         case'Q':scanf(" %d",&p);query(p);break;
         default:scanf(" %d",&p);undo(p);break;
      }
   }
   
// while(1);
   return 0;
}





你可能感兴趣的:(trie,字典树,Scrivener,Tries,ioi2012)