美团2021校招笔试-编程题(通用编程试题,第10场)----公司食堂

小美和小团所在公司的食堂有N张餐桌，从左到右摆成一排，每张餐桌有2张餐椅供至多2人用餐，公司职员排队进入食堂用餐。小美发现职员用餐的一个规律并告诉小团：当男职员进入食堂时，他会优先选择已经坐有1人的餐桌用餐，只有当每张餐桌要么空着要么坐满2人时，他才会考虑空着的餐桌；

当女职员进入食堂时，她会优先选择未坐人的餐桌用餐，只有当每张餐桌都坐有至少1人时，她才会考虑已经坐有1人的餐桌；

无论男女，当有多张餐桌供职员选择时，他会选择最靠左的餐桌用餐。现在食堂内已有若干人在用餐，另外M个人正排队进入食堂，小团会根据小美告诉他的规律预测排队的每个人分别会坐哪张餐桌。

输入描述:

第一行输入一个整数T（1<=T<=10），表示数据组数。

每组数据占四行，第一行输入一个整数N（1<=N<=500000）；

第二行输入一个长度为N且仅包含数字0、1、2的字符串，第i个数字表示左起第i张餐桌已坐有的用餐人数；

第三行输入一个整数M（1<=M<=2N且保证排队的每个人进入食堂时都有可供选择的餐桌）；

第四行输入一个长度为M且仅包含字母M、F的字符串，若第i个字母为M，则排在第i的人为男性，否则其为女性。

输出描述:

每组数据输出占M行，第i行输出一个整数j（1<=j<=N），表示排在第i的人将选择左起第j张餐桌用餐。

输入例子1:

1
5
01102
6
MFMMFF

输出例子1:

2
1
1
3
4
4

#include<iostream>
#include<cmath>
#include<iomanip>
#include<algorithm>
using namespace std;
int main()
{
    int T;
    cin>>T;
    int N;
    cin>>N;
    char aa;
    int a[500500];
    int x0 = 0;
    int x1 = 0;
    int x2 = 0;
    for(int i = 1; i <= N; i++)
    {
        cin>>aa;
        if(aa == '0') {x0++;a[i] = 0;}
        else if(aa=='1') {x1++;a[i] = 1;}
        else {x2++;a[i] = 2;}
//        cout<<a[i]<<" "; 
        
    }
//    cout<<x0<<" "<<x1<<" "<<x2;
    int M;
    cin>>M;
    char cc[100500];
    int re[100500];
    for(int i = 0; i < M; i++)
    {
        char c;
        cin>>c;
        if(c == 'M')//男孩子 
        {
            if(x1 == 0)  
            {
                for(int j = 1; j <= N; j++)
                {
                    if(a[j]==0)
                    {
                        x1++;
                        x0--;
                        re[i] = j;
                        a[j]++;
                        break;
                    }
                }
            }
            else if(x1!=0)//优先1的
            {
                for(int j = 1; j <= N; j++)
                {
                    if(a[j]==1)
                    {
                        x1--;
                        x2++;
                        a[j]++;
                        re[i] = j;
                        break;
                    }
                }
            }
//            cout<<re[i]<<"aaa";
        }
        if(c == 'F')//女孩子 
        {
            if(x0==0)  
            {
                for(int j = 1; j <= N; j++)
                {
                    if(a[j]==1)
                    {
                        x1--;
                        x2++;
                        re[i] = j;
                        a[j]++;
                        break;
                    }
                }
            }
            else if(x0!=0)//优先0的
            {
                for(int j = 1; j <= N; j++)
                {
                    if(a[j]==0)
                    {
                        x1++;
                        x0--;
                        a[j]++;
                        re[i] = j;
                        break;
                    }
                }
            }
        }
    }
    for(int i = 0; i < M; i++)
    {
        cout<<re[i]<<endl;
    }
    return 0;
}

过了10%，我觉得是超时了，毕竟暴力真的很费时