二分——解方程

二分——解方程

题目描述

给出n个整数和x，请问这n个整数中是否存在三个数a，b，c使得ax2+bx+c=0，数字可以重复使用。

输入描述

第一行两个整数n，x
第二行n个整数a[i]表示可以用的数
1 <= n <= 1000, -1000 <= a[i], x <= 1000

输出描述

YES表示可以
NO表示不可以

示例

输入

2 1
1 -2

输出

YES

方法一：二分

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int n,x,a[1005];
	cin>>n>>x;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	sort(a,a+n);
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			int l=0,r=n-1,mid;
			while(l<=r){
				mid=l+(r-l)/2;
				if(a[i]*x*x+a[j]*x+a[mid]==0){
					cout<<"YES"<<endl;
					return 0;
				}
				else if(a[i]*x*x+a[j]*x+a[mid]<0){
					l=mid+1;
				}
				else{
					r=mid-1;
				}
			}
		}
	}
	cout<<"NO"<<endl;
	return 0;
}

方法二：map 映射

#include<bits/stdc++.h>
using namespace std;
map<int,int> mp;

int main()
{
	int n,x,a[1005];
	cin>>n>>x;
	for(int i=0;i<n;i++){
		cin>>a[i];
		mp[a[i]]=1;
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			int ans=a[i]*x*x+a[j]*x;
			if(mp[-ans]){
				cout<<"YES"<<endl;
				return 0;
			}
		}
	}
	cout<<"NO"<<endl;
	return 0;
}