美团2021秋招——公司食堂

链接：https://www.nowcoder.com/questionTerminal/601815bea5544f389bcd20fb5ebca6a8
来源：牛客网
 

小美和小团所在公司的食堂有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

输出

2
1
1
3
4
4

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
int findM(int* arr, int len){
    int pos = -1;
    for(int i = 0; i < len; i++){
        if(arr[i] == 1){
            pos = i+1;
            break;
        }
    }
    if(pos == -1){
        for(int i = 0;i < len; i++){
            if(arr[i] == 0){
                pos = i+1;
                break;
            }
        }
    }
    return pos;
}
int findF(int* arr , int len){
    int pos = -1;
    for(int i = 0;i < len; i++){
            if(arr[i] == 0){
                pos = i+1;
                break;
            }
        }
    if(pos == -1){
        for(int i = 0; i < len; i++){
            if(arr[i] == 1){
                pos = i+1;
                break;
            }
        }
    }
    return pos;
}
struct cafeteria{
    int n;
    int* nums;
    int waiting;
    char* str;
};
int main(){
    int t;
    scanf("%d",&t);
    struct cafeteria input[t];
    int k = 0;
    int i = 0;
    char* temp =  (char*)malloc(sizeof(char) * 1000000);
    while(scanf("%s",temp) != EOF){
        switch(k){
            case 0:
                input[i].n = atoi(temp);
                input[i].nums = (int*)malloc(sizeof(int) * input[i].n);
                k++;
                break;
            case 1:
                for(int j = 0; j < input[i].n; j++){
                    input[i].nums[j] = temp[j] - '0';
                }
                k++;
                break;
            case 2:
                input[i].waiting = atoi(temp);
                input[i].str = (char*)malloc(sizeof(char) * (input[i].waiting+1));
                k++;
                break;
            case 3:
                strcpy(input[i].str,temp);
                k = 0;
                i++;
                break;
        }
    }
    //input end
    for(int i = 0; i < t; i++){
        int* sit = (int*)malloc(sizeof(int) * input[i].waiting);
        for(int j = 0; j < input[i].waiting; j++){
            if(input[i].str[j] == 'M'){
                sit[j] = findM(input[i].nums, input[i].n);
                input[i].nums[sit[j]-1]++;
            }else{
                sit[j] = findF(input[i].nums, input[i].n);
                input[i].nums[sit[j]-1]++;
            }
        }
        for(int k = 0; k < input[i].waiting; k++){
            printf("%d\n",sit[k]);
        }
    }
}