算法训练 自行车停放

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

　　有n辆自行车依次来到停车棚，除了第一辆自行车外，每辆自行车都会恰好停放在已经在停车棚里的某辆自行车的左边或右边。(e.g.停车棚里已经有3辆自行车，从左到右编号为：3,5,1。现在编号为2的第4辆自行车要停在5号自行车的左边，所以现在停车棚里的自行车编号是：3,2,5,1)。给定n辆自行车的停放情况，按顺序输出最后停车棚里的自行车编号。

输入格式

　　第一行一个整数n。
　　第二行一个整数x。表示第一辆自行车的编号。
　　以下n-1行，每行3个整数x,y,z。
　　z=0时，表示编号为x的自行车恰停放在编号为y的自行车的左边
　　z=1时，表示编号为x的自行车恰停放在编号为y的自行车的右边

输出格式

　　从左到右输出停车棚里的自行车编号

样例输入

4
3
1 3 1
2 1 0
5 2 1

样例输出

3 2 5 1

数据规模和约定

　　n<=100000
　　自行车编号为不超过100000的正整数。

 

代码超时 80

原因主要是在n非常大的情况下，在遍历整个可变长数组的过程中很费时

#include<iostream>
#include<vector>
//anthor:ZQ
using namespace std;
int main(){
    vector<int>obj;
    vector<int>::iterator it;
    int n,ns,nq,flag,n1;
    cin>>n;
    cin>>n1;
    obj.push_back(n1);
    for(int i=1;i<n;i++){
        cin>>ns>>nq>>flag;
        for(it=obj.begin();it!=obj.end();it++){
            if(*it==nq){
                if(flag==0){
                    obj.insert(it,ns);
                }else{
                    obj.insert(it+1,ns);
                }
                break; 
            }
        }
    }
    for(it=obj.begin();it!=obj.end();it++){
        cout<<*it<<" ";
    }
    return 0;
} 

改进代码

不用去一个一个去寻找自行车位置，调用<algorithm>库中的函数find（）直接锁定位置

代码：

#include<iostream>
#include<vector>
#include<algorithm>
//anthor:ZQ
using namespace std;
int main(){
    vector<int>obj;
    vector<int>::iterator it;
    vector<int>::iterator position;
    int n,ns,nq,flag,n1;
    cin>>n;
    cin>>n1;
    obj.push_back(n1);
    for(int i=1;i<n;i++){
        cin>>ns>>nq>>flag;
        position=find(obj.begin(),obj.end(),nq);
        if(flag==0){
            obj.insert(position,ns);
        }else{
            obj.insert(position+1,ns);
        }
    }
//        for(it=obj.begin();it!=obj.end();it++){
//            if(*it==nq){
//                if(flag==0){
//                    obj.insert(it,ns);
//                }else{
//                    obj.insert(it+1,ns);
//                }
//                break; 
//            }
//        }
    for(it=obj.begin();it!=obj.end();it++){
        cout<<*it<<" ";
    }
    return 0;
}