leetcode 1386.安排电影院座位

原创于 2022-09-15 14:40:27 发布 · 445 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

本文介绍了一种通过二进制表示和位运算技巧，高效计算在给定座位预留条件下最多能安排多少家庭的方法。使用C++代码实现，重点讲解了如何利用哈希表存储和判断排位占用情况。

一、题解
对于每一排座位，被安排家庭数目有四种情况：
1、安排两个家庭，即列2到9都没被预约。
2、安排一个家庭在左边，即2到5没被预约。
3、安排一个家庭在中间，即4到7没被预约。
4、安排一个家庭在右边，即6到9没被预约。
由此可以发现对于每一排的第1列和第10列，无论是否被占用，都不影响最后答案，而且对于第2、第3和第四种情况，都可以合并来计算。
对于该题考虑二进制的位运算来解决，将第2到9列设置为一个8位的二进制数，如果被占用，则置为1，否则置为0。对于第2到第9列未被占用的排，则按第1种情况乘以2作为安排家庭数。排对应的8个二进制数用哈希表来存储，未存储说明没有座位被预约。
二、C++代码

class Solution {
public:
    int maxNumberOfFamilies(int n, vector<vector<int>>& reservedSeats) {
        int left=0b11110000,middle=0b11000011,right=0b00001111;
        
        unordered_map<int,int> occ;
        for(vector<int>& seat:reservedSeats){
            if(seat[1]>=2&&seat[1]<=9){
                occ[seat[0]]|=(1<<(seat[1]-2));
            }
        }
        int ans = (n-occ.size())*2;
        for(auto& [r,bit]:occ){
            if(((bit|left)==left)||((bit|middle)==middle)||(bit|right)==right){
                ans++;
            }
        }
        return ans;
    }
};