「コードフォース」D. 無限セット

D. 無限集合

https://codeforces.com/contest/1635/problem/D

トピックの説明

さまざまな正の整数と無限のセット S で構成される配列があるので、セット S をすべて$x$条件の数。

$x$の条件(いずれか 1 つを満たすことができます):

1. $バツ=a_{}(1\le 私\le n)$
2. $バツ=2歳+1\left(y\epsilon S\right)$（$y\; は$S セット内になければなりません)
3. $バツ=4年\left(年\epsilon S\right)$ ($y$ は S セットにある必要があります)

無限集合 S で$2^p$の数。

説明を入力

最初の行には、2 つの整数 n と p (1≤n,p≤2⋅10 ⁵ ) が含まれます。

第二行包含 n 个整数 a1,a2,…,an (1≤ai≤10⁹)。

保证 a 中的所有数字都是不同的。

输出描述

打印一个整数，即 S 中严格小于 2p 的元素数。 请记住以 10⁹+7 为模打印。

样例

#1

2 4
6 1

9

#2

4 7
20 39 5 200

14

#3

2 200000
48763 1000000000

448201910

提示

在第一个示例中，小于 2⁴ 的元素是 {1,3,4,6,7,9,12,13,15}。

在第二个例子中，小于 2⁷ 的元素是 {5,11,20,23,39,41,44,47,79,80,83,89,92,95}。

解析

有一说一，二进制的题目我确实没写过…，一时间也只能想到暴力，可这样不可能过，后面看大佬题解，发现要用二进制…

**Tips：**以后但凡看到这种 $2^p$ 次幂的形式，要往二进制想想。

根据二进制，$x$ 的第二和第三条件可以转为：

1. 第一条件其实就是 A 数组的所有元素均属于 S
2. $2x+1$ 相当于 $2<<1$ 然后加 $1$ （向后添加一个 1）
3. $4x$ 相当于 $4<<2$ （向后添加两个 0）

题目要我们求 S 中所有小于 $2^p$ 的数，其实就是求对应的二进制最高位 1 的后面 0 的变化，例如 $p=3$ 时，对应二进制为 $1000$ ，我们可以得到且不保证符合的有 $0000、0100、0110、0111、0010、0011、0001$ 。

通过上面的例子，我们发现，只是 1 后面的 0 发生了变化，而我们的规则是要么增加一个 1，要么增加两个0（这0是一起添加的，不能分开）。

我们假设一个数为 $x$ ，那么我们可以得到：

$x$ 增加位数	得到的不同的数
0	$x$	1
1	$x1$	1
2	$x00、x11$	2
3	$\times 001、\times 100、\times 111$	3
4	$\times 0000、\times 1100、\times 0011、\times 1111、\times 1001$	5

上の表の意味: 数値の 2 進数の後に桁数が続き、毎回 1 か 2 つの 0 しか追加できないため、最終的にいくつの異なる数値が得られるかを示します。

そして、それらがどのように変化しても、一意であることが保証されていることがわかります。

これはフィボナッチ数列であることがわかります$F\left[i\right]=F[i-1]+F[i-2]$、ここでは数値が増加することを意味します$i$ビットで取得できる異なる数の数は$F\left[私\right]$。

今は桁数を追加するための数値を求めているだけです.異なる数値を取得した場合は、前の数値をカウントしないため、接頭辞と接頭辞 [ i ] = 接頭辞 [ i − 1 ]を作成する必要があります。$prefix\left[i\right]=prefix[i-1]+F\left[i\right]$の場合、定義は$prefix\left[i\right]$は増加を意味します$i$桁の後に得られるすべての異なる数字

最後に、数値ai a_iの場合、$a_{}$Aj a_j は別の数値から取得できます$a_{}$、二重にカウントしないように、$a_{}$たとえば、A には 2 と 8 の 2 つの要素があり、8 は 2 を介して取得できるため、2 だけを保持する必要があります。

2：0010

8：1000

0010 は 1 回変更できます (つまり、最後に 2 つの 0 を追加します) ==> 0010 00 ==> 10 00

ACコード

public class Main {
    
    
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static StreamTokenizer st = new StreamTokenizer(br);
    static PrintWriter out = new PrintWriter(new BufferedOutputStream(System.out));

    public static void main(String[] args) throws Exception {
    
    
        final int MOD = 1000000000 + 7;
        final int MAX = 200005;
        int n = nextInt(), p = nextInt(), ans = 0;
        int[] A = new int[MAX];
        int[] F = new int[MAX];
        int[] prefix = new int[MAX];
        
        F[0] = F[1] = 1;
        prefix[0] = 1; prefix[1] = 2;
        HashMap<Integer, Boolean> map = new HashMap<>();

        for(int i = 2; i < MAX; i++) {
    
    
            F[i] = (F[i-1] + F[i-2]) % MOD;
            prefix[i] = (prefix[i-1] + F[i]) % MOD;
        }

        for(int i = 1; i <= n; i++) {
    
    
            A[i] = nextInt();
            map.put(A[i], true);
        }
	    
        // 对数组元素进行去重
        for(int i = 1; i <= n; i++) {
    
    
            int x = A[i];
            while(x != 0) {
    
    
                if(x % 2 == 1) x >>= 1; // 条件2，末尾加 1
                else if(x % 4 == 0) x >>= 2; // 条件3，末尾加两个 0
                else break;
                if(map.containsKey(x)) {
    
     // 判断 x 是否已经存在 S 中
                    map.remove(A[i]); // 若存在，则表示 A[i] 可以通过其它元素得到，估抛弃该元素
                    break;
                }
            }
        }
        
        Set<Integer> set = map.keySet();
        for(Integer x : set) {
    
    
            int bit = 32 - Integer.numberOfLeadingZeros(x); // 获取 x 二进制位数的前导零个数
            if(bit > p) continue;
            ans += prefix[p - bit];
            ans %= MOD;
        }
        
        System.out.println(ans);
    }
    
    public static int nextInt() throws Exception {
    
    
        st.nextToken();
        return (int) st.nval;
    }
}