题目描述
小 X 捡到了一台复读机，这台复读机可以向机器人发号施令。机器人将站在一棵完全二叉树的根上，完全二叉树是无限延伸的。你将向复读机录入一串指令，这串指令单个字符可以是：
L：命令机器人向当前节点的左子走；R：命令机器人向当前节点的右子走；U：命令机器人向当前节点的父亲走（若没有，则命令非法）。
录入指令后，复读机将会把指令无限复读下去。比如命令为 LR，那么机器人会遵从 LRLRLRLR... 一直走下去。
这棵完全二叉树上有一个 nnn 个节点的连通块，保证这个连通块包含根节点。连通块上的每个节点都埋有宝藏，机器人到达过的地方如果有宝藏，则会将其开采。如果一个地方没有宝藏，机器人也可以到那里去。机器人也可以反复经过一个地方。
显然，这个连通块本身也是一棵二叉树。
现在，有人告诉了小 X 埋有宝藏的这棵二叉树的前序遍历，小 X 需要寻找到一条尽量短的指令，使得机器人能够挖掘出所有宝藏。
输入格式
一行一个字符串，由 0123 中的字符组成，表示埋有宝藏的这棵二叉树的前序遍历。
0：表示这是一个没有儿子的节点。1：表示这是一个只有左子的节点。2：表示这是一个只有右子的节点。3：表示这是一个既有左子又有右子的节点。
输出格式
一个整数，表示最短指令的长度。
输入输出样例
输入 #1
1313000
输出 #1
3
输入 #2
333003003300300
输出 #2
15
说明/提示
【样例 1 说明】
一种可行的最短指令为 LRU
- Subtask 1（31 points）：2≤n≤10^2
- Subtask 2（32 points）：2≤n≤200
- Subtask 3（37 points）：无特殊限制。
对于 100%100\%100% 的数据，2≤n≤2×10^3
学业繁忙，题解先欠着。。。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 struct Tree_ 8 { 9 int l,r,size; 10 }Tree[5005],c[5005]; 11 int dep[5005],pa[5005],n,num,ans; 12 char s[5005]; 13 void build(int x,int fa) 14 { 15 dep[x]=dep[fa]+1;Tree[x].size=1; 16 pa[x]=fa; 17 if (s[x-1]=='0') return; 18 if (s[x-1]=='1') 19 { 20 Tree[x].l=x+1; 21 build(x+1,x); 22 Tree[x].size+=Tree[Tree[x].l].size; 23 } 24 if (s[x-1]=='2') 25 { 26 Tree[x].r=x+1; 27 build(x+1,x); 28 Tree[x].size+=Tree[Tree[x].r].size; 29 } 30 if (s[x-1]=='3') 31 { 32 Tree[x].l=x+1; 33 build(x+1,x); 34 Tree[x].size+=Tree[Tree[x].l].size; 35 Tree[x].r=x+Tree[x].size; 36 build(x+Tree[x].size,x); 37 Tree[x].size+=Tree[Tree[x].r].size; 38 } 39 //cout<<x<<' '<<Tree[x].l<<' '<<Tree[x].r<<endl; 40 } 41 int merge(int rt,int now,int goal) 42 { 43 if (!now) return rt; 44 if (!rt) rt=++num; 45 c[rt].size=1; 46 if (now==goal) return rt; 47 c[rt].l=merge(c[rt].l,Tree[now].l,goal); 48 c[rt].size+=c[c[rt].l].size; 49 c[rt].r=merge(c[rt].r,Tree[now].r,goal); 50 c[rt].size+=c[c[rt].r].size; 51 return rt; 52 } 53 int find(string opt) 54 {int i; 55 int now=1,lst=0,len=opt.size(); 56 while (now) 57 { 58 lst=now; 59 for (i=0;i<len;i++) 60 { 61 if (opt[i]=='L') now=Tree[now].l; 62 else now=Tree[now].r; 63 if (!now) break; 64 } 65 merge(1,lst,now); 66 } 67 return c[1].size; 68 } 69 void dfs(int x,string opt) 70 {int cnt=0; 71 if (!x) return; 72 if (x!=1) 73 { 74 memset(c,0,sizeof(c)); 75 num=1; 76 cnt=find(opt); 77 ans=min(ans,2*(cnt-1)-dep[x]+1); 78 } 79 dfs(Tree[x].l,opt+'L'); 80 dfs(Tree[x].r,opt+'R'); 81 } 82 int main() 83 { 84 cin>>s; 85 n=strlen(s); 86 build(1,0); 87 ans=2e9; 88 dfs(1,""); 89 cout<<ans; 90 }