占座位（C++实训题）

写在前面：仅为个人代码/总结，未必标准，仅供参考！如有错误，还望指出交流，共同进步！

占座位

【问题描述】sun所在学校的教室座位每天都是可以预占的。
一个人可以去占多个座位，而且一定是要连续的座位，如果占不到他所要求的这么多座位，那么他就一个座位也不要了。为了降低难度，每次分配座位按座位号从小到大查找，采用最先适配法分配座位。

【输入形式】输入有多组数据。
每组数据输入座位排数n，0<n<=100（座位的排列数相等，座位是按每行从左到右依次排序的,，第1行的最右边一个座位与第二行的第一个座位视为连续座位），m（ 0<m<=min(100,n*n) ）个人。
然后输入k（0<k<=100），最后输入k个命令。
命令只有两种：
1.in id num（代表id,0<=id<m,要占num个座位，若占不到连续的num(0<num<=20)个座位表示该命令无效）
2.out id（代表id要释放他之前占的所有座位）
注意：如果id之前占过座还没释放那么之后他的in命令都是无效的，
如果id之前没占过座位那么他的out命令也是无效的。

【输出形式】对每个in命令输出yes或者no，如果命令有效则输出yes，无效则输出no。
在yes no后面只带有回车，不带其他任何字符。

【样例输入】
4 10
9
in 1 7
in 2 3
in 3 3
in 3 3
in 4 3
out 2
in 5 6
out 3
in 5 6

【样例输出】
yes
yes
yes
no
yes
yes
no
yes
yes

【示例代码】

#include <bits/stdc++.h>
using namespace std;
struct Seat
{
    int id;
    int length;//所占座位的连续长度
    bool cur;//记录座位的当前状态
};
int main()
{
    int n,m;
    cin>>n>>m;
    int yu=n*n;//yu用于实时存储座位队尾的空闲座位数
    vector <Seat> P;
    Seat temp;
    int k;
    cin>>k;
    while(k--)
    {
        string s;
        int id,num;
        cin>>s>>id;
        if(s=="in") {cin>>num;}
        if(s=="in")
        {
            int len=P.size();
            bool flag=false;
            if(len!=0)
            {
                bool b=false;
                for(int j=0;j<len;j++)
                {
                    if(P[j].id==id&&P[j].cur==true)
                    {
                        cout<<"no"<<endl;
                        b=true;
                        break;
                    }
                }
                if(b==true) {continue;}
                for(int j=0;j<len;j++)
                {
                    if(P[j].cur==false&&P[j].length>=num)//首次出现满足所占座位长度条件，则占最前面的座位（已占过但被释放了的座位）
                    {
                        temp.id=id;
                        temp.cur=true;
                        temp.length=num;
                        P[j].length-=n;//实时变化该段座位的状态和长度
                        if(P[j].length==0)//若该段座位刚好再次被全占
                        {
                            vector<Seat>::iterator it=P.begin()+j;
                            P.erase(it);
                        }
                        P.insert(P.begin()+j,temp);
                        cout<<"yes"<<endl;
                        flag=true;
                        break;
                    }
                }
            }
            if(flag==false)
            {
                 if(yu>=num)
                {
                    yu-=num;
                    temp.id=id;
                    temp.length=num;
                    temp.cur=true;
                    P.push_back(temp);
                    cout<<"yes"<<endl;
                }
                else {cout<<"no"<<endl;}
            }
        }
        else if(s=="out")
        {
            int len=P.size();
            bool flag=false;
            for(int j=0;j<len;j++)
            {
                if(P[j].id==id)
                {
                    if(P[j].cur==false)
                    {
                        break;
                    }
                    P[j].cur=false;
                    //如果释放的座位前后存在已被释放的座位，则合并该段空闲的座位
                    if(j>0)
                    {
                        if(P[j-1].cur==false)
                        {
                            P[j-1].length+=P[j].length;
                            vector<Seat>::iterator it=P.begin()+j;
                            P.erase(it);
                        }
                    }
                    if(j<len-1)
                    {
                        if(P[j+1].cur==false)
                        {
                            P[j].length+=P[j+1].length;
                            vector<Seat>::iterator it=P.begin()+j+1;
                            P.erase(it);
                        }
                    }
                    flag=true;
                    break;
                }
            }
            if(flag==false) {cout<<"no"<<endl;}
            else {cout<<"yes"<<endl;}
        }
    }
    return 0;
}