【杭电】[2141]Can you find it?

文章字数:243

问题描述

问题分析

因为N最大为500
所以如果开三重循环进行判断的话会超时
那么可以思考如何进行优化

Ai+Bj+Ck=X

Ai+Bj=X-Ck

所以可以预处理Ai+Bj 保存之后进行排序 然后对于每个询问 循环C然后在预处理中 二分查找有无X-Ck

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#include<stdio.h>
#include<algorithm>
using namespace std;
int l,n,m,cnt;
int a[520],b[520],c[520],res[520*520];
bool find(int x) {
	int l=0,r=cnt-1;
	while(l<=r) {
		int mid=(l+r)/2;
		if(res[mid]>x)
			r=mid-1;
		else if(res[mid]<x)
			l=mid+1;
		else
			return true;
	}
	return false;
}
int main() {
	int kase=0;
	while(scanf("%d %d %d",&l,&n,&m)!=EOF) {
		printf("Case %d:\n",++kase);
		for(int i=0; i<l; i++)
			scanf("%d",&a[i]);
		for(int i=0; i<n; i++)
			scanf("%d",&b[i]);
		for(int i=0; i<m; i++)
			scanf("%d",&c[i]);
		cnt=0;
		for(int i=0; i<l; i++)
			for(int j=0; j<n; j++)
				res[cnt++]=a[i]+b[j];
		sort(res,res+cnt);
		int T;
		scanf("%d",&T);
		while(T--) {
			int t;
			scanf("%d",&t);
			int i;
			for(i=0; i<m; i++) {
				if(find(t-c[i]))
					break;
			}
			if(i<m)
				printf("YES\n");
			else
				printf("NO\n");
		}
	}
	return 0;
}

该内容采用 CC BY-NC-SA 4.0 许可协议。

如果对您有帮助或存在意见建议,欢迎在下方评论交流。

加载中...