[Codeforces #379 F. Anton and School]位运算技巧

[Codeforces #379 F. Anton and School]位运算技巧

1. 题目链接

[Codeforces #379 F. Anton and School]

2. 题意描述

给定$n$，以及两个长度为$n$的数组$b[]$, 和数组$c[]$,找出一个数组$a[]$满足：

$$\begin{cases} b_i = \sum_{j=1}^{j=n}(a_i \space\space \text{&} \space\space a_j) \\ c_i = \sum_{j=1}^{j=n}(a_i\space\space \text{ | } \space\space a_j) \end{cases}$$
如果不存在，输出”-1”;存在就输出数组 $a[]$。 $(1\le n \le 200000, 0\le b_i,c_i\le 10^9)$

3. 解题思路

这道题目技巧性很强。
假设结果存在。
首先，需要利用这个公式(非常重要)：

$$\color{red}{(x\text{&}y)+(x\text{|}y)=x+y}$$
所以：

$$\therefore b_i+c_i=\sum_{j=1}^{n}(a_i+a_j)=n*a_i+\sum_{j=1}^{n}a_j\\ \therefore 令SUM=\sum_{i=1}^{i=n}(a_i+b_i)=2n*\sum_{j=1}^{n}a_j\\ \therefore 对于\forall i,(1\le i\le n), 可求得:\\ a_i=\frac{b_i+c_i-\sum_{j=1}^{n}a_j}{n}\\[3ex] =\frac{b_i+c_i-\frac{SUM}{2*n}}{n}\\ $$
所以，可以在 $O(N)$的复杂度下求出数组 $a[]$， 并且 $a[]$是唯一的。
接下来，就是判断 $a[]$是否合法。即通过数组 $a[]$,来生成数组 $b^{'}[], c^{'}[]$【技巧】, 然后判断 $b$与$b^{'}[]$、 $c$与$c^{'}[]$是否相等。
数组 $a[]$中每一项 $a_i$的第 $k$位，记做$A_{i,k}$。
求出数组 $a_1,a_2,\ldots,a_n$，在第 $k$位$1$的个数，记做 $cnt_k$。即 $cnt_k=\sum_{i=1}^{n}A_{i,k}$。
令:
$$\begin{cases} B_{i,k}=\sum_{j=1}^{n}(A_{i,k}\space\space\text{&}\space\space A_{j,k})\\ C_{i,k}=\sum_{j=1}^{n}(A_{i,k}\space\space\text{ | }\space\space A_{j,k}) \end{cases}$$
那么，
$$B_{i,k}= \begin{cases} 0,& A_{i,k}=0\\ cnt_k,& A_{i,k}=1\\ \end{cases}\\ C_{i,k}= \begin{cases} cnt_k,& A_{i,k}=0\\ 0,& A_{i,k}=1\\ \end{cases}$$
那么，
$$b^{'}_{i}=\sum_{k=0}^{bits}(B_{i,k}*2^k)\\ c^{'}_{i}=\sum_{k=0}^{bits}(C_{i,k}*2^k)\\$$ .
这个技巧就在于将一个数字拆成多个二进制位进行考虑。
最后的复杂度为 $O(N*bits)$。

4. 实现代码

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef pair<int, int> PII;

const int MX = 200000 + 5;
const int BITS = 32;

int n;
vector<LL> a, b, c;
bool check() {
    vector<int> cnt(BITS);
    for(int k = 0; k < BITS; k++) {
        cnt[k] = 0;
        for(int i = 0; i < n; i++) {
            cnt[k] += (a[i] >> k) & 1;
        }
    }
    vector<vector<int> > B(n), C(n);
    for(int i = 0; i < n; i++) {
        B[i].resize(BITS);
        C[i].resize(BITS);
        for(int k = 0; k < BITS; k++) {
            if((a[i] >> k) & 1) {
                B[i][k] = cnt[k];
                C[i][k] = n;
            } else {
                B[i][k] = 0;
                C[i][k] = cnt[k];
            }
        }
    }
    for(int i = 0; i < n; i++) {
        int bi = 0, ci = 0;
        for(int k = 0; k < BITS; k++) {
            bi += B[i][k] * (1 << k);
            ci += C[i][k] * (1 << k);
        }
        if(bi != b[i] || ci != c[i]) return false;
    }
    return true;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("input.txt", "r", stdin);
#endif // ONLINE_JUDGE
    ios::sync_with_stdio(false);
    cin >> n;
    a.resize(n);
    b.resize(n);
    c.resize(n);
    for(int i = 0; i < n; i++) cin >> b[i];
    for(int i = 0; i < n; i++) cin >> c[i];
    if(n == 1) {
        cout << (b[0] == c[0] ? b[0] : -1) << endl;
        return 0;
    }
    LL sumA = 0;
    for(int i = 0; i < n; i++) {
        sumA += c[i] + b[i];
    }
    if(sumA % (2LL * n)) {
        cout << -1 << endl;
        return 0;
    }
    sumA /= (2LL * n);
    for(int i = 0; i < n; i++) {
        a[i] = b[i] + c[i] - sumA;
        if(b[i] == 0 && a[i] != 0) {
            cout << -1 << endl;
            return 0;
        }
        if(a[i] % n || a[i] < 0) {
            cout << -1 << endl;
            return 0;
        }
        a[i] /= n;
    }
    if(check() == false) {
        cout << -1 << endl;
        return 0;
    }
    for(int i = 0; i < n; i++) {
        cout << a[i] << " \n"[i == n - 1];
    }
    return 0;
}