Leetcode 1711. 大餐计数（桶）

1711. 大餐计数

难度中等83收藏分享切换为英文接收动态反馈

大餐 是指 恰好包含两道不同餐品 的一餐，其美味程度之和等于 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

太久没做题了，一开始写了排序+二分枚举2的次幂结果T了，后来突然发现可以用桶给卡过去。。。1e6的数组开在函数里memset20次丝毫不慌

int countPairs(vector<int>& deliciousness) {
	const int mod = 1000000007;
	int ans = 0;
	sort(deliciousness.begin(), deliciousness.end());
  //注释部分是二分写法
	// int sz = deliciousness.size();
	// for(int i = 1; i <= (1 << 23); i *= 2) {
	// 	for(int j = 0; j < sz; j++) {
	// 		int y = i - deliciousness[j];
	// 		//cout << deliciousness[j] << " " << y << endl;
	// 		int pos1 = lower_bound(deliciousness.begin() + j + 1, deliciousness.end(), y) - deliciousness.begin() - j - 1;
	// 		int pos2 = upper_bound(deliciousness.begin() + j + 1, deliciousness.end(), y) - deliciousness.begin() - j - 1;
	// 		pos1 += j + 1;
	// 		pos2 += j + 1;
	// 		if(y < deliciousness[0] || y > deliciousness[sz - 1]) continue;
	// 		if(pos1 == sz) continue;
	// 		if(deliciousness[pos1] != y) continue;//注意这里有可能越界
	// 		ans = (ans + pos2 - pos1) % mod;//pos2可能为sz
	// 	}
	// }
	// return ans
	int sz = deliciousness.size();
	for(int i = 1; i <= (1 << 22); i *= 2) {
		int b[1048599];
		memset(b, 0, sizeof(b));
		for(int j = 0; j < sz; j++) {
			int y = i - deliciousness[j];
			if(y > 1048599 || y < 0) continue;
			ans = (ans + b[y]) % mod;
			b[deliciousness[j]]++;
		}
	}
	return ans;
}