抽签游戏

题意：

        你的朋友提议玩一个游戏：将写有数字的n个纸片放入口袋中，你可以从口袋中抽取4次纸片，每次记下纸片上的数字后都将其放回口袋。如果这4个数字的和是m，就是你赢，否则就是你的朋友赢。请编写一个程序，纸片所写数据为：K1,K2,K3,K4.....Kn，判断你是否有赢的可能性。

（限制条件：1<=n<=50，1<=m<=10⁸，1<=K1...Kn<=10⁸）

直接暴力循环：

代码：

int n,m,data[55];
	while(~scanf("%d%d",&n,&m))
	{
		for(int i=0;i<n;i++)
			scanf("%d",&data[i]);

		bool Judge = false;

		for(int a=0;a<n;a++)
			for(int b=0;b<n;b++)
				for(int c=0;c<n;c++)
					for(int d=0;d<n;d++)
						if(data[a] + data[b] + data[c] + data[d] == m)
							Judge = true;
		if(Judge)
			printf("Yes\n");
		else
			printf("No\n");
	}

结果：

（对于第一组数据来说：1+3+3+3 = 10，第二组数据中不存在和为9的方案）

        一般竞赛的时间限制为1s。对于循环来说：10万次绰绰有余，100万次勉强，1000万就很危险了。

        题目要求是n<=50。上面算法的复杂度为n⁴。循环次数为50*50*50*50 = 625 0000。情况不是很好。

那么问题来了，怎么优化呢？

        先想一下：一次一次加的话，还不如最后直接匹配。例如，抽出了3个数，那么第四次抽出的如果不可能是m-k1-k2-k3，那么就不能赢。时间复杂度为：n³log(n)。

        这里查询建议用二分法查询，比较快，C++有相关查询函数binary_search()，这样就不能用数组了，需要用vector。

代码：

int n,m,data;
	vector<int> v;
	while(~scanf("%d%d",&n,&m))
	{
		for(int i=0;i<n;i++)
		{
			scanf("%d",&data);
			v.push_back(data);
		}

		bool Judge = false;

		for(int a=0;a<n;a++)
			for(int b=0;b<n;b++)
				for(int c=0;c<n;c++)
				{
					int temp = m - v[a] - v[b] - v[c];
					Judge = binary_search(v.begin(),v.end(),temp);
				}
		if(Judge)
			printf("Yes\n");
		else
			printf("No\n");
	}

结果：

        这样的话，那能否计算任意两个数的值，新建一个序列，再去查找新序列中是否存在两个数的和等于m呢？这样时间复杂度为：n²log(n)。

优化最终代码：

 int n,m,data;
	vector<int> v;
	while(~scanf("%d%d",&n,&m))
	{
		v.clear();
		for(int i=0;i<n;i++)
		{
			scanf("%d",&data);
			v.push_back(data);
		}
		// 创建新序列
		vector<int>vTemp;
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				vTemp.push_back(v[i]*v[j]);

		bool Judge = false;
		for(int i=0;i<n*n;i++)
		{
			int temp = m - vTemp[i];
			if(binary_search(v.begin(),v.end(),temp))
				Judge = true;
		}
		if(Judge)
			printf("Yes\n");
		else
			printf("No\n");
	}

结果：

尽量减少循环的嵌套次数，然后是循环次数。

PS：最近看的竞赛书籍是《挑战程序设计竞赛》，都是高手的经验总结。下载网址：http://www.gaofi.cn/book/detail/402