求每个cluster中box的个数

nudt_qxx

于 2024-03-07 11:12:28 发布

阅读量200

点赞数 2

分类专栏： Leetcode/剑指offer/算法/数据结构 C++ 文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/xiangxianghehe/article/details/136528292

版权

C++ 同时被 2 个专栏收录

157 篇文章 8 订阅

订阅专栏

Leetcode/剑指offer/算法/数据结构

84 篇文章 7 订阅

订阅专栏

该C++代码通过队列算法处理一组Box，判断它们之间的交集形成簇，统计每个簇中Box的数量。

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

题目描述
有一堆box，若干个box有交集就是一个簇，求每个cluster中box的个数。注意：可以自定义box的结构体，无需考虑旋转角度。
请用C+实现上述代码

#include <vector>
#include <queue>
#include <algorithm>

struct Box {
    int x1, y1, x2, y2;
};

bool isIntersect(const Box& a, const Box& b) {
    return !(a.x2 < b.x1 || a.x1 > b.x2 || a.y2 < b.y1 || a.y1 > b.y2);
}

int main() {
    std::vector<Box> boxes = {/* 输入box数据 */};
    std::vector<std::vector<int>> clusters;
    std::vector<bool> visited(boxes.size(), false);

    for (int i = 0; i < boxes.size(); ++i) {
        if (visited[i]) continue;

        std::queue<int> q;
        q.push(i);
        visited[i] = true;

        std::vector<int> cluster;
        while (!q.empty()) {
            int j = q.front();
            q.pop();
            cluster.push_back(j);

            for (int k = 0; k < boxes.size(); ++k) {
                if (visited[k] || !isIntersect(boxes[j], boxes[k])) continue;
                q.push(k);
                visited[k] = true;
            }
        }

        clusters.push_back(cluster);
    }
    for (const auto& cluster : clusters) {
        std::cout << "Cluster size: " << cluster.size() << std::endl;
    }
    return 0;
}