CF-Round#641-div2-D题

CF-Round#641-div2-D题

D. Orac and Medians

传送门

这道题是贪心题~

题目大意：给你一个长度为n的数组，和一个数字k，问你是否可以通过无限次的操作把这个数组中的所有元素全部都变成k。
操作如下：
选取某一段区间[l, r]，把这个区间中的所有元素都变成这个区间的中位数。
中位数的定义：当前区间的长度为len,中位数是这个区间的第len + 1 / 2向下取整的元素。

如果可以，输出yes，否则输出no

本题思路：最容易想到的特殊情况。如果这个数组中不存在k这个数，那么我们就直接输出no；如果这个数组的长度只有1，并且正好为k时，输出yes。
其余情况。
我们对数组a[]进行以下操作：
遍历数组a[]：
如果a[i] > k，那我们把a[i]置为2，即a[i] = 2;
如果a[i] < k，那我们把a[i]置为0，即a[i] = 0;
如果a[i] = k，那我们把a[i]置为1，即a[i] = 1;
我们只需要遍历一下长度为2或者长度为3的区间中是否存在两端元素都大于等于k的情况。即判断a[l]不为0且a[r]不为0；
这样我们就可以把这个区间中的所有元素都变成较小的元素。
依次这样下去可以把所有的元素都变成k。
如果区间长度为2或者为3的都不存在两端元素大于等于k的情况。那么说明无法把所有元素变为k。（因为取中位数的标准是向下取整的，都取得那个较小的那个数）
所以如果判断到长度为2或者长度为3的两端元素全都大于等于k的情况的时候，我们就可以输出yes了。

代码部分：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;

int a[N];

int solve()
{
	int flag = 0;
	int n, k;
	scanf ("%d%d", &n, &k);
	for (int i = 0; i < n; i++)
	{
		scanf ("%d", &a[i]);
		if (a[i] > k)
		{
			a[i] = 2;
		}
		else if (a[i] < k)
		{
			a[i] = 0;
		}
		else
		{
			a[i] = 1;
			flag = 1;
		}
	}
	if (!flag)
	{
		return 0;
	}
	if (n == 1)
	{
		return 1;
	}
	for (int i = 0; i < n; i++)
	{
		for (int j = i + 1; j < n && j - i <= 2; j++)
		{
			if (a[i] && a[j])
			{
				return 1;
			}
		}
	}
	return 0;
}

int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		int flag = solve();
		flag ? cout << "yes\n" : cout << "no\n";
	}
	return 0;
}