LeetCode -- 1711. 大餐计数

import java.util.HashMap;
import java.util.Map;

/**
 * 1711. 大餐计数
 *
 * 方法：枚举2的幕
 * 容斥原理
 * 使用哈希表记录重复
 * 
 */

public class Solution1711 {
  // 余数
  private final int remainder = 1000000007;
  // 根据题意，最大为2的22次幂
  private final int maxNum = 1 << 22;
  public int countPairs(int[] deliciousness) {
    Map<Integer, Integer> map = new HashMap<>();
    int ans = 0;
    for (int x : deliciousness) {
      int i = 1;
      while (i < maxNum) {
        int t = i - x;
        if (map.containsKey(t)) {
          ans += map.get(t);
          if (ans >= remainder) ans -= remainder;
        }
        i <<= 1;
      }
      map.put(x, map.getOrDefault(x, 0) + 1);
    }
    return ans;
  }

  // 获取大于等于自身的最小的2次幂，hashmap源码
  // 并未用到
  private int to2power(int num) {
    num--;
    num |= num >> 1;
    num |= num >> 2;
    num |= num >> 4;
    num |= num >> 8;
    num |= num >> 16;
    return num + 1;
  }

  public static void main(String args[]) {
    Solution1711 solution1711 = new Solution1711();
    int res = solution1711.countPairs(new int[]{1,1,1,3,3,3,7});
    System.out.println(res);
  }
}