火车入轨_算法

 1 #include
 2 using namespace std;
 3 const int MAXSIZE=100;
 4 void main()
 5 {
 6     int n;
 7     cin>>n;
 8     int a[MAXSIZE],b[MAXSIZE];
 9     int stack[MAXSIZE];
10     for(int i=0;i)
11     {
12         a[i]=i+1;
13         cin>>b[i];                   　　 //出栈顺序
14     }
15     int top=-1;
16     int count=0;
17     int i=0;
18     for(;;)
19     {
20         if(i<n)
21         {
22             ++top;
23             stack[top]=a[i++];            //入栈
24             cout<<"PUSH"<<endl;
25         }
26     
27         if(stack[top]==b[count])
28         {
29             top--;count++;
30             cout<<"POP"<<endl;
31         }
32         else if(i==n)
33         {
34             cout<<"NO"<<endl;
35             break;
36         }
37         if(count==n)
38         {
39             cout<<"YES"<<endl;
40             break;
41         }
42         if(top<-1)
43         {    
44             cout<<"NO"<<endl;
45             break;
46         }
47     }
48 
49 }

 1 #include
 2 using namespace std;
 3 const int MAXN=1000+10;
 4 int n,target[MAXN];
 5 void main()
 6 {
 7     while(cin>>n)
 8     {
 9         int stack[MAXN],top=0;
10         int A=1,B=1;                                          　　  //A用来记录入栈次数，B用来记录出轨的火车序号
11         for(int i=1;i<=n;i++)
12             cin>>target[i];                                        //记录出轨顺序
13         int ok=1;
14         while(B<=n)
15         {
16             if(A==target[B]){A++;B++;}
17             else if(top && stack[top]==target[B]){top--;B++;}    　　//出栈
18             else if((A<=n)) stack[++top]=A++;                   　　 //入栈
19             else {ok=0;break;}
20         }
21         if(ok)
22             cout<<"Yes"<<endl;
23         else
24             cout<<"No"<<endl;
25     }

 1 /*STL栈来实现*/
 2 #include
 3 #include
 4 using namespace std;
 5 const int MAXN=1000+10;
 6 int n,target[MAXN];
 7 int main()
 8 {
 9     while(cin>>n)
10     {
11         stack<int> s;
12         int A=1,B=1;
13         for(int i=1;i<=n;i++)
14             cin>>target[i];
15         int ok=1;
16         while(B<=n)
17         {
18             if(A==target[B]){A++;B++;}
19             else if(!s.empty() && s.top()==target[B]){s.pop();B++;}
20             else if(A<=n) s.push(A++);
21             else {ok=0;break;}
22         }
23         if(ok)
24             cout<<"YES"<<endl;
25         else
26             cout<<"NO"<<endl;
27     }
28 }