使用无符号整数作为二进制在c++中实现集合

#include <iostream>

// 定义集合的最大元素个数
const int MAX_SIZE = 32;

// 使用无符号整数实现集合
class Set {
private:
    unsigned int data;

public:
    Set() : data(0) {}

    // 添加元素到集合
    void add(int element) {
        if (element >= 0 && element < MAX_SIZE) {
            data |= (1 << element);
        }
    }

    // 从集合中移除元素
    void remove(int element) {
        if (element >= 0 && element < MAX_SIZE) {
            data &= ~(1 << element);
        }
    }

    // 检查集合是否包含某个元素
    bool contains(int element) {
        if (element >= 0 && element < MAX_SIZE) {
            return (data & (1 << element)) != 0;
        }
        return false;
    }
};

int main() {
    Set set;
    set.add(1);
    set.add(3);
    set.add(5);

    std::cout << "Set contains 1: " << set.contains(1) << std::endl;
    std::cout << "Set contains 2: " << set.contains(2) << std::endl;
    std::cout << "Set contains 3: " << set.contains(3) << std::endl;
    std::cout << "Set contains 4: " << set.contains(4) << std::endl;
    std::cout << "Set contains 5: " << set.contains(5) << std::endl;

    return 0;
}