【Leetcode】1386. Cinema Seat Allocation

最新推荐文章于 2024-10-01 05:04:32 发布

记录算法题解

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量258

点赞数

分类专栏： LC 栈、队列、串及其他数据结构文章标签： leetcode 算法 c++ 数据结构

本文链接：https://blog.csdn.net/qq_46105170/article/details/129109626

版权

LC 栈、队列、串及其他数据结构专栏收录该内容

717 篇文章 8 订阅

订阅专栏

该问题是一个关于位运算和数组处理的算法题目。给定一个电影院的座位布局，使用位运算来标记已占用的座位，然后通过分情况讨论计算最多可以安排多少个四人组。首先统计不同状态的行，最后加上未被占用行的数量的两倍作为结果。时间复杂度为O(lv)，其中lv是输入数组的元素数量。

摘要由CSDN通过智能技术生成

题目地址：

https://leetcode.com/problems/cinema-seat-allocation/description/

在这里插入图片描述
电影院有 $n$ 行，每行 $10$ 个位置，给定若干已经占据的位置，问能安排多少个 $4$ 人组，每个组占据 $2345, 4567, 6789$ 之一。

先用位运算表示一下有占据位置的行的情况，然后分情况讨论来累加，最后加上 $2$ 乘以空行的数量。代码如下：

class Solution {
 public:
  int maxNumberOfFamilies(int n, vector<vector<int>>& v) {
    unordered_map<int, int> mp;
    for (auto& e : v) mp[e[0]] |= 1 << e[1] - 1;
    int res = 0;
    const int full = (1 << 4) - 1;
    for (auto& [k, x] : mp) {
      if (!(x >> 1 & full) && !(x >> 5 & full))
        res += 2;
      else if (!(x >> 1 & full) || !(x >> 3 & full) || !(x >> 5 & full))
        res++;
    }

    res += 2 * (n - mp.size());
    return res;
  }
};