最初、「これは本当に AGC か!??」となってた
問題概要
以上 以下の整数の中からいくつか選んで、OR 和をとってできる値が何通りあるか求めよ。
制約
考えたこと
一見するとこどふぉっぽい見た目の問題だけど、実はすごく AGC っぽい感じの問題だった。
とりあえず、最上位桁から見て行って、 と とで一致しているところは無視できるので、全部 にしてしまって除外しておこう。そうすると、ある桁 が存在して
- の 桁目は (B にとっての最上位桁)
- の 桁目は
という状態にできる。よって、次の場合分けがしたくなる。
- OR 和の 桁目が の場合 ( 〜 の場合に帰着)
- OR 和の 桁目が の場合 ( から までフルに使える)
このように場合分けすると、いかにも再帰的に解くような流れになっていきそうで、こどふぉ感を最初感じた。
d 桁目が 0 のとき
まずは OR 和の 桁目が の場合を考えてみる。この場合は
の OR 和で作られる数を求めたい。そして当然ながら、
はすべて作れる!!そしてよくよく考えると、これらの数以外は作れない。なぜなら、一般に整数 に対して他の正の整数 と OR をとると
OR
が成立するからだ。つまり、 からいくつか選んだ OR 和が 未満になることはありえない。一方、 桁目が 1 になることもないので、作れる数は 以上 以下の整数ということで確定する。
d 桁目が 1 のとき
桁目が 1 であるようなものを少なくとも 1 つ以上含めないとダブルカウントになってしまうので注意が必要そう。でも、 以上 以下の整数の中に、 ( 桁目までがすべて 0) が含まれるので、そんなに心配しなくて大丈夫。
問題は としたとき、
- 以上 以下の整数
- 以上 以下の整数
を使って作られる整数をすべて求める問題となる。 の最高位の桁を とすると、さっきと同じ理屈から、
- 以上 以下の整数からは、 以上 以下の整数がすべて作れる
- 以上 以下の整数からは、 以上 ( とする) 以下の整数がすべて作れる
ということになる。前者の値と後者の値の OR 和も考えないといけないが、それは完全に前者に含まれる。よって、
- のときは、 個がすべて作れる
- のときは、 個が作れる
ということになる。
コード
以上をまとめると、 で解ける。
#include <bits/stdc++.h> using namespace std; long long solve(long long A, long long B) { if (A == B) return 1; int d = 61; for (; d >= 0; --d) { if ((B & (1LL<<d)) && !(A & (1LL<<d))) break; A &= ~(1LL<<d); B &= ~(1LL<<d); } long long zero = (1LL<<d) - A; // d 桁目が 0 int d2 = d-1; for (; d2 >= 0; --d2) if (B & (1LL<<d2)) break; long long ue = 1LL<<(d2+1); long long one = 0; if (A <= ue) one = (1LL<<d); else one = (1LL<<d) - A + ue; return zero + one; } int main() { long long A, B; cin >> A >> B; cout << solve(A, B) << endl; }