Tsinghua dsa pa2

第一题，列车调度(train)

在这个题目中，模拟压栈出栈，同时判断调度方案是否可行。

 #include <cstdio>
 #include <cstring>
 #define MAX 1600005
 using namespace std;
 int stack[MAX];//stack[0]作为旗标
 int top;
 int pushed;//已经压入的最大值
 int out[MAX];//输出
 char result[*MAX][];
 int result_len;//result的长度
 int main(int argc,char *argv[]){
 //    freopen("in.txt","r",stdin);
     int len,capacity;
     scanf("%d %d", &len, &capacity);
     for(int k=;k<len;k++)
         scanf("%d",&(out[k]));
     int i=;//出栈的元素的下标
     while(i<len){//当i==len时，所有out中的元素都成功出栈
         if(stack[top]==out[i]){
             strcpy(result[result_len++],"pop");
             top--;
             i++;
         }else if(stack[top]<out[i]){
             while(pushed<out[i]){
                 stack[++top]=++pushed;
                 if(top>capacity){
                     printf("No\n");
                     return ;
                 }
                 strcpy(result[result_len++],"push");
             }
         }else{
                 printf("No\n");
                 return ;
         }
     }
     for(int j=;j<result_len;j++)
         printf("%s\n",result[j]);
     return ;
 }

第二题，隧道（Tunel），通过95%

这道题的关键在于如何在O(1)时间内获取队列中的最大值。提示见邓俊辉老师的数据结构习题解答讲义。

 #include <cstdio>
 using namespace std;
 #define MAX 2000005

 int height[MAX];
 int head,tail;

 typedef struct{
     int index;
     int count;
 }Mirrorqueue;
 Mirrorqueue que[MAX];
 int headq,tailq;

 int main(int argc, char *argv[]){
 //    freopen("in.txt","r",stdin);
     int n;
     scanf("%d\n", &n);
     char c;
     while(n--){
         scanf("%c", &c);
         int counttmp=;
         int i=;
         switch (c){
         case 'E':
             scanf("%d\n", height+tail);
             for(i=tailq;i>headq;i--){
                 if(height[que[tailq-].index]<height[tail]){
                     counttmp+=que[tailq-].count;
                     tailq--;
                 }else if(height[que[tailq-].index]==height[tail]){
                     que[tailq-].count+=counttmp;
                     break;
                 }else{
                     que[tailq].index=tail;
                     que[tailq].count=counttmp;
                     tailq++;
                     break;
                 }
             }
             if(i==headq){
                 que[tailq].index=tail;
                 que[tailq].count=counttmp;
                 tailq++;
             }
             tail++;
             break;
         case 'M':
             scanf("%c",&c);//忽略换行符
             printf("%d\n",height[que[headq].index]);
             break;
         case 'D':
             scanf("%c",&c);//忽略换行符
             que[headq].count--;
             if(que[headq].count==)
                 headq++;
             printf("%d\n",height[head++]);
         }
     }
     return ;
 }

第三题，真二叉树重构（Rebuild），通过95%

提示：

 #include <cstdio>
 using namespace std;
 #define MAX 4000005
 int preord[MAX],postord[MAX];

 int stack[];
 int top;

 int printed,nodeNums;//已经打印的结点数目和总的结点数目
 int search(int *source,int head,int tail,int e){
     for(int i=head;i<=tail;i++){
         if(source[i]==e)
             return i;
     }
 }
 int rebuild(int *pre, int *post,int preh,int pret,int posth,int postt){
     if(preh==pret){
         if(printed==nodeNums-){
             printf("%d\n",pre[preh]);
             printed++;
         }
         else{
             printf("%d ",pre[preh]);
             printf("%d ",stack[--top]);
             printed+=;
         }
         return ;
     }
     //store pre[preh];//根
     stack[top++]=pre[preh];
     int lroot=search(post,posth,postt,pre[preh+]);//在后序遍历中查找左子树根
     int rroot=search(pre,preh,pret,post[postt-]);
     rebuild(pre,post,preh+,rroot-,posth,lroot);
     rebuild(pre,post,rroot,pret,lroot+,postt-);
     //输出
     return ;
 }
 int main(int argc, char *argv[]){
 //    freopen("in.txt","r",stdin);
     int nodeNum;
     char c;
     scanf("%d\n",&nodeNum);
     nodeNums=nodeNum;
     for(int i=;i<nodeNum;i++)
         scanf("%d",preord+i);
     scanf("%c",&c);//忽略换行符
     for(int j=;j<nodeNum;j++)
         scanf("%d",postord+j);
     rebuild(preord,postord,,nodeNum-,,nodeNum-);
     return ;
 }