Div. 2 Codeforces Round #822 A-F( 二 )


时间复杂度 \(O(n^2)\)
空间复杂度 \(O(n^2)\)
代码#include <bits/stdc++.h>#define ll long longusing namespace std;int a[357][357], b[357];bool solve() {int n;cin >> n;for (int i = 1;i <= n;i++) cin >> b[i];for (int i = 1;i <= n;i++) {for (int j = 1;j <= n;j++) {a[i][j] = ((i * (j - i) + b[i]) % n + n) % n;}}for (int i = 1;i <= n;i++) {for (int j = 1;j <= n;j++) {cout << a[i][j] << ' ';}cout << '\n';}return true;}int main() {std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);int t = 1;//cin >> t;while (t--) {if (!solve()) cout << -1 << '\n';}return 0;}F题解知识点:记忆化搜索,线性dp,数学 , 位运算 。
先是一个结论:定义函数 \(parity(a)\) 表示 \(a\) 二进制位 \(1\) 的个数的奇偶性(奇数返回 \(1\) ,偶数返回 \(0\)),那么 \(S_i = parity(i)\)。
证明非常简单:

  1. 由于 \(S\) 的生成方法是每次都从原来的一份取反得到 \(S'\) 放到 \(S\) 末尾,所以第 \(k(k\geq 1)\) 次扩展后 \(S\) 的编号一定是 \([0,2^{k-1}]\)。
  2. 对于第 \(k+1\) 次新生成的 \(S'\) 中的每一位编号 \(i'\),满足 \(i’ = i + 2^k\) ,因为编号 \(i\) 的第 \(k\) 位之前一定是 \(0\),所以这次变换实际上是将编号 \(i\) 的第 \(k\) 位变为 \(1\) 作为编号 \(i'\) 。
  3. 显然,所有数字都是从编号 \(0\) 开始数次变换得到的,每次变换都会将编号的一位 \(0\) 变为 \(1\),因此我们记录二进制 \(1\) 的数量就能得知这个编号从 \(0\) 变换了多少次 。
  4. \(S_0 = 0\) ,所以编号 \(i\) 有偶数个 \(1\) 就是变了偶数次,故 \(S_i=0\) ,否则 \(S_i = 1\)。即 \(S_i = parity(i)\)。
有了这个结论,我们就可以对问题进行量化 。记原问题答案为 \(f(n,m)\),有 \(f(n,m) = \sum_{i = 0}^{m-1} [parity(i) \neq parity(n+i)]\)。
当 \(m = 0\) 时 , 显然有 \(f(n,0) = 0\)。
当 \(m\) 为奇数时,先对末尾判断再对 \(m-1\) 讨论(偶数讨论方便一点) , 有 \(f(n,m) = f(n,m-1) + [parity(i) \neq parity(n+i)]\)。
当 \(m\) 为偶数时:
  • \(n\) 为偶数,有如下关系:
    \[\begin{aligned} && &parity(2k) \neq parity(n+2k) \\ &\Leftrightarrow& &parity(2k+1) \neq parity(n+2k+1)\\\end{aligned}\]因为偶数末尾总是 \(0\)  , 加 \(1\) 不会影响其余的二进制位,所以 \(1\) 的数量明确加 \(1\) ,奇偶性一定同时改变 。
    \[\begin{aligned} && &parity(2k) \neq parity(n+2k) \\ &\Leftrightarrow& &parity(k) \neq parity(\frac{n}{2}+k)\end{aligned}\]因为偶数末尾总是 \(0\),删去这个 \(0\) 后 , 数字奇偶性不变 。
    那么有如下公式:
    \[\begin{aligned} f(n,m) &= \sum_{i = 0}^{m-1} [parity(i) \neq parity(n+i)]\\ &= 2\sum_{k = 0}^{\frac{m}{2}-1} [parity(2k) \neq parity(n+2k)]\\ &= 2\sum_{k = 0}^{\frac{m}{2}-1} [parity(k) \neq parity(\frac{n}{2}+k)]\\ &= 2f(\frac{n}{2},\frac{m}{2})\end{aligned}\]
  • \(n\) 为奇数,有如下关系:
    \[\begin{aligned} && &parity(2k) \neq parity(n+2k) \\ &\Leftrightarrow& &parity(2k) = parity(n+2k-1)\\ &\Leftrightarrow& &parity(k) = parity(\frac{n-1}{2}+k)\\\end{aligned}\]以及,
    \[\begin{aligned} && &parity(2k+1) \neq parity(n+2k+1) \\ &\Leftrightarrow& &parity(2k) = parity(n+2k+1)\\ &\Leftrightarrow& &parity(k) = parity(\frac{n+1}{2}+k)\\\end{aligned}\]证明同上 。
    \[\begin{aligned} f(n,m) &= \sum_{i = 0}^{m-1} [parity(i) \neq parity(n+i)]\\ &= \sum_{k = 0}^{\frac{m}{2}-1} ([parity(2k) = parity(n+2k-1)] + [parity(2k) = parity(n+2k+1)])\\ &= \sum_{k = 0}^{\frac{m}{2}-1} ([parity(k) = parity(\frac{n-1}{2}+k)] + [parity(k) = parity(\frac{n+1}{2}+k)])\\ &= m - f(\frac{n-1}{2},\frac{m}{2}) - f(\frac{n+1}{2},\frac{m}{2})\end{aligned}\]
至此,我们就可以通过记忆化搜索进行求解了 。
时间复杂度 \(O(\log n \log m)\)
空间复杂度 \(O(\log n \log m)\)
代码#include <bits/stdc++.h>#define ll long longusing namespace std;bool check(ll a, ll b) {return __builtin_parityll(a) != __builtin_parityll(b);}map<pair<ll, ll>, ll> mp;ll f(ll n, ll m) {if (m == 0) return 0;if (mp.count({ n,m })) return mp[{n, m}];if (m & 1) return mp[{n, m}] = f(n, m - 1) + check(m - 1, n + m - 1);if (n & 1) return mp[{n, m}] = m - f(n / 2, m / 2) - f((n + 1) / 2, m / 2);else return mp[{n, m}] = 2 * f(n / 2, m / 2);}bool solve() {ll n, m;cin >> n >> m;cout << f(n, m) << '\n';return true;}int main() {std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);int t = 1;cin >> t;while (t--) {if (!solve()) cout << -1 << '\n';}return 0;}

推荐阅读