1711. 大餐计数

package leetcode.editor.cn;
//大餐 是指 恰好包含两道不同餐品 的一餐，其美味程度之和等于 2 的幂。 
//
// 你可以搭配 任意 两道餐品做一顿大餐。 
//
// 给你一个整数数组 deliciousness ，其中 deliciousness[i] 是第 i 道餐品的美味程度，返回你可以用数组中的餐品做出的不同 大
//餐 的数量。结果需要对 109 + 7 取余。 
//
// 注意，只要餐品下标不同，就可以认为是不同的餐品，即便它们的美味程度相同。 
//
// 
//
// 示例 1： 
//
// 
//输入：deliciousness = [1,3,5,7,9]
//输出：4
//解释：大餐的美味程度组合为 (1,3) 、(1,7) 、(3,5) 和 (7,9) 。
//它们各自的美味程度之和分别为 4 、8 、8 和 16 ，都是 2 的幂。
// 
//
// 示例 2： 
//
// 
//输入：deliciousness = [1,1,1,3,3,3,7]
//输出：15
//解释：大餐的美味程度组合为 3 种 (1,1) ，9 种 (1,3) ，和 3 种 (1,7) 。 
//
// 
//
// 提示： 
//
// 
// 1 <= deliciousness.length <= 105 
// 0 <= deliciousness[i] <= 220 
// 
// Related Topics 数组 哈希表 
// 👍 97 👎 0

import java.util.*;

//Java：大餐计数
public class CountGoodMeals {
    public static void main(String[] args) {
        Solution solution = new CountGoodMeals().new Solution();
        // TO TEST
        int[] ints = new int[100000];
        for (int i = 0; i <= 99999; i++) {
            ints[i] = 32;
        }
        System.out.println(solution.countPairs(ints));
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        long mod = (long) 1e9 + 7;

        public int countPairs(int[] deliciousness) {
            long num = 0;
            Arrays.sort(deliciousness);
            int min = deliciousness[0] * 2;
            int max = deliciousness[deliciousness.length - 1] * 2;
            List<Integer> list = getDoubleList(min, max);
            for (Integer integer : list) {
                int left = 0;
                int right = deliciousness.length - 1;
                while (left < right) {
                    int i = deliciousness[left] + deliciousness[right];
                    if (integer > i) {
                        left++;
                    } else if (integer < i) {
                        right--;
                    } else {
                        //这时候就要去看下左右的数有多少个
                        //先去判断一下左右两个数是不是相等
                        //如果相等的话 看有多少个数 如果6个数就是有(5*6)/2种
                        if (deliciousness[left] == deliciousness[right]) {
                            int nums = right - left;
                            num += ((long) nums * (long) (nums + 1)) / 2;
                            break;
                        } else {
                            //判断有多少个跟left、right相等的数
                            int leftNum = getNum(left, deliciousness);
                            int rightNum = getNum(right, deliciousness);
                            //有leftNum*rightNum个数
                            num += leftNum * rightNum;
                            //还需要继续遍历
                            left += leftNum;
                            right -= rightNum;
                        }
                    }
                }
            }
            return (int) (num % mod);
        }

        /**
         * 获取min~max中间的 2^n的数
         *
         * @param min
         * @param max
         * @return
         */
        private List<Integer> getDoubleList(int min, int max) {
            List<Integer> list = new ArrayList<>();
            int n = min;
            n = n / 2 * 2;
            if (n < 2) {
                list.add(1);
                n = 2;
            }
            if (n < min) n = n * 2;
            while (n <= max) {
                list.add(n);
                n = n * 2;
            }
            return list;
        }

        /**
         * 获取数组种与ints[index]值相等的个数
         *
         * @param index
         * @param ints
         * @return
         */
        private int getNum(int index, int[] ints) {
            int val = ints[index];
            int result = 0;
            for (int i = index; i >= 0; i--) {
                if (ints[i] == val) {
                    result++;
                } else {
                    break;
                }
            }
            if (index + 1 < ints.length) {
                for (int i = index + 1; i < ints.length; i++) {
                    if (ints[i] == val) {
                        result++;
                    } else {
                        break;
                    }
                }
            }
            return result;
        }


    }
}

执行用时：
33 ms
, 在所有 Java 提交中击败了
97.29%
的用户

思路：
先对数组进行排序，确认最小与最大美味层度。在最小与最大的美味程度之间找2^n的数。接着在排序完的数组中利用双指针法查找数(leetcode第一题两数之和解法)
ps：原本的用hash做，后面测试用例跑不过，在公司回家的路上通过两个数相加联想到lc第一题，记录下。。。