算法问题

试题描述

你的朋友提议玩一个游戏：将写有数字的n 个纸片放入口袋中，你可以从口袋中抽取4 次纸片，每次记下纸片上的数字后都将其放回口袋中。如果这4 个数字的和是m，就是你赢，否则就是你的朋友赢。你挑战了好几回，结果一次也没赢过，于是怒而撕破口袋，取出所有纸片，检查自己是否真的有赢的可能性。请你编写一个程序，判断当纸片上所写的数字是k1,k2, …, kn 时，是否存在抽取4 次和为m 的方案。

输入

第一行为两个整数n，m；第二行为n个整数k1,k2, …, kn 。

输出

如果存在，输出“Yes”；否则，输出“No”。

输入示例

3 10 1 3 5

输出示例

Yes

#include<iostream>
#include <algorithm>
using namespace std;
int main(){
	int n,m;
	int i,j,k,l,sum=0; 
	int a[100];
	scanf("%d%d",&n,&m);
	for(i=0;i<n;i++){
		scanf("%d",&a[i]);
	}//数据存放
	sort(a,a+n);//数据排序
	for(i=0;i<n;i++){
		sum+=a[i];
		if(sum>=m){
			sum=0;
			break;
		}
		for(j=0;j<n;j++){
			sum+=a[j];
			if(sum>=m){
				break;			
			}
			for(k=0;k<n;k++){
				sum+=a[k];
				if(sum>=m){
					break;
				}
				for(l=0;l<n;l++){
					sum+=a[l];
					if(sum>m){
						break;
					}
					if(sum==m){
						cout<<"YES\n"<<endl; 
						return 0;
					}
						sum-=a[l];
					}
					sum-=a[k];
			}
			sum-=a[j];
		}	
		sum=0; 
	} 

	cout<<"NO"<<endl;
	return 0;
}

看到网上关于这道题的csdn博客，大都直接用了四个循环，这样子时间复杂度为n的四次方，这样的算法肯定是不行的。于是思考了一种优化方式，先进行排序，然后选择是否要进入下一层训话，这样子，时间复杂度有所降低，但是不稳定，不知道还有没有更好的优化方法，希望可以学习，欢迎大家留言评论