您的位置:首页 > 其它

【杭电2141】Can you find it?

2016-07-26 20:42 239 查看




我首先想到的是三重循环—-超时,后来想前两个数组双重循环套第三个数组二分查找—-还是超时,再后来就是现在的代码思路了300多ms。

其实和第二种想法很接近了,就是二分的对象变了,对第三个数组遍历循环,对前两个数组各项求和得到的数组求二分,因为这样就大大大大减少了循环次数,二分是省时间的嘛。

本题难点算法优化。

#include<cmath>
#include<algorithm>
#include<stdio.h>
using namespace std;
int a[505],b[505],c[505];
int dp[250025];
int main() {
int i,j,k,m,n,s,t,cas=0,p,flag;
while(scanf("%d%d%d",&k,&m,&n)!=EOF) {
for(i=0; i<k; i++)
scanf("%d",&a[i]);
for(i=0; i<m; i++)
scanf("%d",&b[i]);
for(i=0; i<n; i++)
scanf("%d",&c[i]);
p=0;
for(i=0; i<k; i++)
for(j=0; j<m; j++)
dp[p++]=a[i]+b[j];
sort(dp,dp+p);
printf("Case %d:\n",++cas);
scanf("%d",&t);
while(t--) {
scanf("%d",&s);
flag=0;
for(i=0; i<n; i++) {
int l=0,r=p-1,mid;
while(l<=r) {
mid=(l+r)/2;
if(dp[mid]+c[i]==s) {
flag=1;
break;
} else if(dp[mid]+c[i]<s)
l=mid+1;
else if(dp[mid]+c[i]>s)
r=mid-1;
}
if(flag)
break;
}
if(flag)
printf("YES\n");
else
printf("NO\n");
}
}
return 0;
}


http://acm.hdu.edu.cn/showproblem.php?pid=2141
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  杭电 二分查找 优化