思维+贪心，CF 1210B - Marcin and Training Camp

EQUINOX1

于 2024-09-27 10:52:22 发布

阅读量143

点赞数 1

分类专栏： OJ刷题解题报告文章标签：算法动态规划数学建模

本文链接：https://blog.csdn.net/EQUINOX1/article/details/142588046

版权

OJ刷题解题报告专栏收录该内容

214 篇文章 0 订阅

订阅专栏

一、题目

1、题目描述

2、输入输出

2.1输入

2.2输出

3、原题链接

1210B - Marcin and Training Camp

二、解题报告

1、思路分析

考虑一个数 x 不会比集合内所有数都强，可以注意到什么？

一定有一个数 y 满足 x & y = x

否则，x 和每个数相比总有对方没有的，那么就没办法了

所以，我们先将所有出现次数大于1 的数拿进集合（升序排序，去重即可）

然后遍历剩下的数字 x，如果集合中存在某个数字y 满足 x & y = x，我们就把x 拿进集合

2、复杂度

时间复杂度： O(nlogn)空间复杂度：O(n)

3、代码详解

#include <bits/stdc++.h>

// #define DEBUG

using u32 = unsigned;
using i64 = long long;
using u64 = unsigned long long;

constexpr int P = 1E9 + 7;
constexpr int inf32 = 1E9 + 7;
constexpr i64 inf64 = 1E18 + 7;

void solve() {
    int n;
    std::cin >> n;
    
    std::vector<i64> a(n);
    std::vector<int> b(n);
    for (int i = 0; i < n; ++ i) 
        std::cin >> a[i];    
    for (int i = 0; i < n; ++ i)
        std::cin >> b[i];

    i64 sum = 0;
    std::vector<int> id(n);
    std::iota(id.begin(), id.end(), 0);

    std::ranges::sort(id, [&a](int i, int j) -> bool {
        return a[i] < a[j];
    });

    std::vector<int> st, buf;

    for (int i = 0, j = 0; i < n; ) {
        i64 tot = 0;
        while (j < n && a[id[i]] == a[id[j]])
            tot += b[id[j ++]];
        if (j - i > 1) {
            sum += tot;
            st.insert(st.end(), id.begin() + i, id.begin() + j);
        }
        else {
            buf.insert(buf.end(), id.begin() + i, id.begin() + j);
        }
        i = j;
    }

    for (int i : buf) {
        bool ok = false;
        for (int j : st) {
            if ((a[i] & a[j]) == a[i]) {
                ok = true;
                break;
            }
        }
        if (ok) {
            st.push_back(i);
            sum += b[i];
        }
    }

    std::cout << sum;
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

#ifdef DEBUG
    int cur = clock();
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif

    int t = 1;
    // std::cin >> t;

    while (t--) {
        solve();
    }
#ifdef DEBUG
    std::cerr << "run-time: " << clock() - cur << '\n';
#endif
    return 0;
}