牛客竞赛-解方程

题目链接：

解方程 (nowcoder.com)

大致题意：

给一个整数x和n个整数，问是否存在a * x * x + b * x + c = 0。其中abc是n个整数中选出来的，且可以重复利用。

分析：

暴力双层for，用二分看看 -c 是否存在。

代码：

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010;

int n, x;
int a[N];

bool bsearch(int u)
{
	int l = 0, r = n - 1;
	while (l < r)
	{
		int mid = l + r >> 1;
		if (a[mid] >= u) r = mid;
		else l = mid + 1; 
	}
	if (a[l] == u) return 1;
	else return 0;
}

int main()
{
	scanf("%d%d", &n, &x);
	for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
	
	sort(a, a + n);
	
	for (int i = 0; i < n; i ++ )
	{
		for (int j = 0; j < n; j ++ )
		{
			int ret = a[i] * x * x + a[j] * x;
			if (bsearch(-ret)) 
			{
				puts("YES");
				return 0;
			}
		}
	}
	
	puts("NO");
	
	return 0;
}