(美团笔试)公司食堂

链接：https://www.nowcoder.com/questionTerminal/601815bea5544f389bcd20fb5ebca6a8?answerType=1&f=discussion
来源：牛客网

小美和小团所在公司的食堂有N张餐桌，从左到右摆成一排，每张餐桌有2张餐椅供至多2人用餐，公司职员排队进入食堂用餐。小美发现职员用餐的一个规律并告诉小团：当男职员进入食堂时，他会优先选择已经坐有1人的餐桌用餐，只有当每张餐桌要么空着要么坐满2人时，他才会考虑空着的餐桌；当女职员进入食堂时，她会优先选择未坐人的餐桌用餐，只有当每张餐桌都坐有至少1人时，她才会考虑已经坐有1人的餐桌；无论男女，当有多张餐桌供职员选择时，他会选择最靠左的餐桌用餐。现在食堂内已有若干人在用餐，另外M个人正排队进入食堂，小团会根据小美告诉他的规律预测排队的每个人分别会坐哪张餐桌。

进阶：时间复杂度O(nlogn)\O(nlogn) ,空间复杂度O(n)\O(n)

第一行输入一个整数T（1<=T<=10），表示数据组数。每组数据占四行，第一行输入一个整数N（1<=N<=500000）；第二行输入一个长度为N且仅包含数字0、1、2的字符串，第i个数字表示左起第i张餐桌已坐有的用餐人数；
第三行输入一个整数M（1<=M<=2N且保证排队的每个人进入食堂时都有可供选择的餐桌）；第四行输入一个长度为M且仅包含字母M、F的字符串，若第i个字母为M，则排在第i的人为男性，否则其为女性。

本题首先想到的就是暴力从桌子左边遍历到桌子右边，实现如下：

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int T,N,M;
    cin >> T;
    for(int i = 0;i < T;i++)
    {
        cin >> N;
        vector<int> desks(N, 0);
        string tmp,man;
        cin >> tmp;
        for(int j = 0; j < N;j++)
            desks[j] = (tmp[j] - '0');
        cin >> M;
        cin >> man;
        for(int j = 0;j < M;j++)
        {
            int target;
            if(man[j] == 'M') target = 1;
            else target = 0;
            bool find = false;
            //找左边桌子数位1的位置
            for(int k = 0;k < N;k++)
            {
                if(desks[k] == target)
                {
                    desks[k]++;
                    cout << k+1 << "\n";
                    find = true;
                    break;
                }
            }
            //没找到，找桌子数量位target的位置
            if(!find)
            {
                if(target == 1) target = 0;
                else target = 1;
                for(int k = 0;k< N;k++)
                {
                    if(desks[k] == target)
                    {
                        desks[k]++;
                        cout << k+1 << "\n";
                        break;
                    } 
                }
            }
        }
    }
    return 0;
}

这种方法只能通过8/12用例。题目进阶要复杂度位nlogn。
考虑使用使用两个最小堆来实现。其中一个两个最小堆分别存储桌子上人数为0和1的桌号，桌号为2的桌子不考虑，因为也不会进入答案中。

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int T,N,M;
    cin >> T;
    for(int i = 0;i < T;i++)
    {
        string tmp,man;
        cin >> N;
        cin >> tmp;
        priority_queue<int, vector<int>, greater<int>> desk0;
        priority_queue<int, vector<int>, greater<int>> desk1;
        //初始化，将桌子分类，编号为2的不需要
        for(int j = 0;j < N;j++)
        {
            if(tmp[j] == '1')
                desk1.push(j+1);
            else if(tmp[j] == '0')
                desk0.push(j+1);
        }
        cin >> M;
        cin >> man;
        for(int j = 0;j < M;j++)
        {
            if(man[j] == 'M')
            {
                if(desk1.size())
                {
                    cout << desk1.top() << "\n";
                    desk1.pop();
                }else
                {
                    cout << desk0.top() << "\n";
                    desk1.push(desk0.top());
                    desk0.pop();
                }
            }else
            {
                if(desk0.size())
                {
                    cout << desk0.top() << "\n";
                    desk1.push(desk0.top());
                    desk0.pop();
                }else
                {
                    cout << desk1.top() << "\n";
                    desk1.pop();
                }
            }
        }
    }
    return 0;
}