異或線性基

Yaosicheng124發表於2024-09-22

我們考慮這樣一個問題:給定 \(N\) 個整數 \(A_1,A_2,\dots,A_N\)。求能異或出多少種不同的值。

我們考慮用一個集合 \(S\) 記錄目前能湊出來的數字。

當我們要加入 \(A_i\) 時,如果 \(A_i\not \in S\),則 \(x\oplus A_i(x\in S)\) 一定都不在 \(S\) 中,否則可以透過 \((x\oplus A_i)\oplus x\) 湊出 \(A_i\)。反之,如果 \(A_i\in S\),則 \(x\oplus A_i(x\in S)\) 一定在 \(S\) 中。

所以只要 \(A_i\not \in S\),則 \(|S|\) 就會翻倍。這樣的空間肯定是無法接受的。

但是我們能觀察到這樣最多隻會翻倍 \(\log \max\{A_i\}\) 次,因為經過這麼多次後一定 \(|S|\ge \max\{A_i\}\),所以不可能再有元素 \(\not \in S\)。我們考慮只記錄使 \(|S|\) 翻倍的元素,這樣空間就可接受了。而線性基就是能表示出整個陣列的最小子集。

但是這樣就不能快速判斷 \(x\) 是否在 \(S\) 中,所以考慮對線性基進行變換。

很容易想到 \([X_1,X_2,\dots,X_k]\) 等價於 \([X_1\oplus X_2,X_2,\dots,X_k]\),所以我們用這樣的方式進行變換。

我們對於 \(i=0,1,\dots,\log \max\{A_i\}\) 記錄至多一個最高位為 \(i\) 的元素 \(d_i\)。每次插入元素 \(x\) 時,從高到底列舉每一位,如果當前這一位上為 \(1\),並且 \(d_i=0\)(這裡表示 \(d_i\) 不存在),那麼就令 \(d_i\leftarrow x\),並結束。如果 \(d_i\ne 0\),則令 \(x\leftarrow x\oplus d_i\),這麼做是為了保證 \(x\) 的最高位變得 \(< i\)

而判斷 \(x\) 是否在 \(S\) 中也很簡單,同樣從高到低列舉,如果當前這位是 \(1\) 就異或 \(d_i\)。如果最後 \(x=0\) 則可以被湊出,否則不行。

由於線性基中的元素不能相互異或出來,所以能異或出 \(2^{|d|}-1\) 種。

空間複雜度 \(O(\log \max\{A_i\})\),時間複雜度 \(O(N\log \max\{A_i\})\)

程式碼

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

const int MAXN = 65;

int n;
ll d[MAXN], ans;

void add(ll x) {
  for(ll i = 61; i >= 0; --i) {
    if(x & (1ll << i)) {
      if(d[i]) {
        x ^= d[i];
      }else {
        d[i] = x;
        return;
      }
    }
  }
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n;
  for(int i = 1; i <= n; ++i) {
    ll x;
    cin >> x;
    add(x);
  }
  for(int i = 61; i >= 0; --i) {
    ans = max(ans, (ans ^ d[i]));
  }
  cout << ans;
  return 0;
}

Luogu P3812

題目描述

給定 \(N\) 個數,求能異或出的最大值。

思路

首先對這些數建線性基,接著從高到低列舉每一位,如果 \(d_i\ne 0\)\(ans\) 在這一位上是 \(0\),則令 \(ans\leftarrow ans \oplus x\)​。

空間複雜度 \(O(\log \max\{A_i\})\),時間複雜度 \(O(N\log \max\{A_i\})\)

程式碼

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

const int MAXN = 65;

int n;
ll d[MAXN], ans;

void add(ll x) {
  for(ll i = 61; i >= 0; --i) {
    if(x & (1ll << i)) {
      if(d[i]) {
        x ^= d[i];
      }else {
        d[i] = x;
        return;
      }
    }
  }
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n;
  for(int i = 1; i <= n; ++i) {
    ll x;
    cin >> x;
    add(x);
  }
  for(int i = 61; i >= 0; --i) {
    ans = max(ans, (ans ^ d[i]));
  }
  cout << ans;
  return 0;
}

AT ABC141 F

題目描述

給定 \(N\) 個數 \(A_1,A_2,\dots,A_N\),你要將其分成兩個非空集合,求兩個集合異或和之和的最大值。

思路

我們令 \(X=A_1\oplus A_2\oplus \dots\oplus A_N\)。如果 \(X\) 在某一位上為 \(1\),那麼很明顯兩個集合的異或和中一個是 \(1\),因此答案一定會加上這一位。否則兩個集合在這一位上的異或一定一樣,所以我們要儘可能讓這裡為 \(1\)。而這個可以使用線性基解決。

空間複雜度 \(O(\log \max\{A_i\})\),時間複雜度 \(O(N\log \max\{A_i\})\)

程式碼

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

int n;
ll res, ans, d[60];

void Insert(ll x) {
  for(int i = 59; i >= 0; --i) {
    if((x >> i) & 1) {
      if(!d[i]) {
        d[i] = x;
        return;
      }
      x ^= d[i];
    }
  }
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n;
  for(int i = 1; i <= n; ++i) {
    ll x;
    cin >> x;
    res ^= x;
    Insert(x);
  }
  for(int i = 59; i >= 0; --i) {
    if(!((res >> i) & 1)) {
      ans = max(ans, ans ^ d[i]);
    }else {
      Insert(d[i] ^ (1ll << i));
    }
  }
  cout << ans + (res ^ ans);
  return 0;
}

CF 895 C

題目描述

給定 \(N\) 個數 \(A_1,A_2,\dots,A_N\),求能乘出多少個完全平方數。

思路

我們可以用一個數每個質因子的次冪的奇偶性來表示一個數,而一個完全平方數就是 \(0\),所以這個問題就轉化成了有多少種方法異或出 \(0\)

在異或線性基中的元素是不可能異或出 \(0\) 的,但能夠異或出線性基外的所有元素,所以任何一個不線上性基的元素集合都能異或成 \(0\),所以答案就為 \(2^{N-|d|}-1\)

空間複雜度 \(O(N)\),時間複雜度 \(O(NV)\),其中 \(V=19\)(質數數量)。

程式碼

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

const int MAXN = 100001, MAXP = 19, MOD = int(1e9) + 7, p[MAXP] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67};

int n, a[MAXN], num[71], d[MAXN], ans;
bool op = 1;

int C(int x) {
  int res = 0;
  for(int i = 2; i * i <= x; ++i) {
    while(x % i == 0) {
      res ^= (1 << num[i]);
      x /= i;
    }
  }
  if(x > 1) {
    res ^= (1 << num[x]);
  }
  return res;
}

void Insert(int x) {
  for(int i = MAXP - 1; i >= 0; --i) {
    if((x >> i) & 1) {
      if(!d[i]) {
        d[i] = x, ans--;
        break;
      }else {
        x ^= d[i];
      }
    }
  }
}

int Pow(int a, int b) {
  int res = 1;
  for(; b; a = 1ll * a * a % MOD, b >>= 1) {
    if(b & 1) {
      res = 1ll * res * a % MOD;
    }
  }
  return res;
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  for(int i = 0; i < MAXP; ++i) {
    num[p[i]] = i;
  }
  cin >> n;
  ans = n;
  for(int i = 1, x; i <= n; ++i) {
    cin >> x;
    Insert(C(x));
  }
  cout << ((Pow(2, ans) - 1) % MOD + MOD) % MOD;
  return 0;
}

CF 1902 F

題目描述

給定一顆樹,每個結點都有一個權值 \(A_i\),有 \(Q\) 次詢問,每次詢問僅使用 \(u\)\(v\) 路徑上的數能否湊出 \(k\)

思路

我們對每個點都記錄一個線性基,這個線性基是從當前結點往根節點插入的。接著考慮怎麼把線性基從父親轉移到兒子。

很容易想到父親沒有插入線性基的元素,兒子肯定也不會插入。所以記錄插入線性基的元素,依次列舉並插入即可。

查詢時列舉插入線性基的元素,如果在路徑上就插入。再進行判斷。

由於至多隻有 \(\log \max\{A_i\}\) 個元素,所以空間複雜度 \(O(N\log \max\{A_i\})\),時間複雜度 \(O(N\log \max\{A_i\})\)

程式碼

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

const int MAXN = 200001;

struct xor_linear_basis {
  int d[20];
  xor_linear_basis() {
    fill(d, d + 20, 0);
  }
  bool Insert(int x) {
    for(int i = 20; i >= 0; --i) {
      if((x >> i) & 1) {
        if(!d[i]) {
          d[i] = x;
          return 1;
        }
        x ^= d[i];
      }
    }
    return 0;
  }
  bool In(int x) {
    for(int i = 20; i >= 0; --i) {
      if((x >> i) & 1) {
        x ^= d[i];
      }
    }
    return !x;
  }
}d[MAXN];

int n, a[MAXN], q, f[19][MAXN], dep[MAXN];
vector<int> e[MAXN], ve[MAXN];

void dfs(int u, int fa) {
  if(d[u].Insert(a[u])) {
    ve[u].emplace_back(u);
  }
  for(int v : ve[fa]) {
    if(d[u].Insert(a[v])) {
      ve[u].emplace_back(v);
    }
  }
  dep[u] = dep[fa] + 1, f[0][u] = fa;
  for(int i = 1; i <= 18; ++i) {
    f[i][u] = f[i - 1][f[i - 1][u]];
  }
  for(int v : e[u]) {
    if(v != fa) {
      dfs(v, u);
    }
  }
}

int LCA(int u, int v) {
  if(dep[u] < dep[v]) {
    swap(u, v);
  }
  int d = dep[u] - dep[v];
  for(int i = 18; i >= 0; --i) {
    if((d >> i) & 1) {
      u = f[i][u];
    }
  }
  if(u == v) {
    return u;
  }
  for(int i = 18; i >= 0; --i) {
    if(f[i][u] != f[i][v]) {
      u = f[i][u], v = f[i][v];
    }
  }
  return f[0][u];
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n;
  for(int i = 1; i <= n; ++i) {
    cin >> a[i];
  }
  for(int i = 1, u, v; i < n; ++i) {
    cin >> u >> v;
    e[u].emplace_back(v);
    e[v].emplace_back(u);
  }
  dfs(1, 0);
  cin >> q;
  for(int i = 1, u, v, k; i <= q; ++i) {
    cin >> u >> v >> k;
    int l = LCA(u, v);
    xor_linear_basis res;
    for(int x : ve[u]) {
      if(dep[x] >= dep[l]) {
        res.Insert(a[x]);
      }
    }
    for(int x : ve[v]) {
      if(dep[x] >= dep[l]) {
        res.Insert(a[x]);
      }
    }
    cout << (res.In(k) ? "YES\n" : "NO\n");
  }
  return 0;
}

相關文章