目录
一、题目
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、代码详解
cpp
#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;
}