抽签

题意

n张写有数字的卡片，有放回地抽4次；
存在4次的数字之和为m的方案输出Yes，否则输出No

思路

思路一：

暴力穷举：O(n^4)

思路二：

利用求和预处理的和数组、二分搜索优化穷举的后两个循环：O(n^2logn)

思路三：

1. 预处理。n个数字两两求和，取矩阵的“上三角”，得到n个数两两之间和的数组km。
2. 排序。对和的数组km排序。
3. 查找。遍历和的数组，并在和的数组中折半查找两个和为m的方案。

总结

• 两个基本意识：二分搜索、数据求和的预处理
• 将四个数分成两两考虑

代码

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;

int main(void)
{
	int n,m;
	scanf("%d%d",&n,&m);
	int *k = new int[n];
	for(int i=0;i<n;++i)
		scanf("%d",&k[i]);
	
	vector<int> km;
	int in=0;
	for(int i=0;i<n;i++)
		for(int j=i;j<n;j++)
			km.push_back(k[i]+k[j]);
	sort(km.begin(),km.end());
	
	bool judge = false;
	for(auto i:km)
	{
		int leftC=0,rightC=km.size()-1;
		while(leftC<=rightC)
		{
			int midC = (leftC+rightC)/2;
			if(i+km[midC]>m)
				rightC = midC;
			else if(i+km[midC]<m)
				leftC = midC;
			else
			{
				judge = true;
				break;
			}
		}
		if(judge==true)
			break;
	}
	if(judge==true)
		printf("Yes");
	else
		printf("No");
		
	return 0;
}