AcWing 1969. 品种邻近

AcWing 1969. 品种邻近

题目描述
农夫约翰的 N 头奶牛排成一排，每头奶牛都用其品种 ID 进行描述。

如果两头相同品种的牛靠得太近，它们就会吵架。

具体的说，如果同一品种的两头奶牛在队列中的位置相差不超过 K，我们就称这是一对拥挤的牛。

请计算品种 ID 最大的拥挤奶牛对的品种 ID。

输入格式
第一行包含两个整数 N 和 K。

接下来 N 行，每行包含一个整数表示队列中一头奶牛的品种 ID。

输出格式
输出品种 ID 最大的拥挤奶牛对的品种 ID。

如果不存在拥挤奶牛队，则输出 −1。

数据范围
1≤N≤50000,
1≤K<N,
品种 ID 范围 [0,10⁶]。

输入样例：

6 3
7
3
4
2
3
4

输出样例：

4

样例解释
一对品种 ID 为 3 的奶牛以及一对品种 ID 为 4 的奶牛属于拥挤奶牛对。

所以，最大拥挤奶牛对的品种 ID 为 4 。

解题思路
啊啦啦啦啦，不用脑子的快乐题，写一下排序比较函数就可
输出最大ID，所以ID降序，同ID则按队列号排好
然后每只牛牛只与前者匹配看是否为拥挤对即可，如果不是自然也不用考虑更远队列号奶牛

代码

#include<iostream>
#include<algorithm>
using namespace std;
struct Cow {
	int id;
	int index;
}cow[50000];
bool cmp(Cow cow1, Cow cow2) {
	if (cow1.id == cow2.id) {
		return cow1.index > cow2.index;
	}
	else{
		return cow1.id > cow2.id;
	}
}
int main() {
	int n, k;
	cin >> n >> k;
	for (int i = 0; i < n; ++i) {
		cin >> cow[i].id;
		cow[i].index = i;
	}
	sort(cow, cow + n,cmp);
	int maxid = -1;
	for (int i = 1; i < n; ++i) {
		if (cow[i-1].id == cow[i].id&&cow[i-1].index-cow[i].index<=k) {
			maxid = cow[i].id;
			break;
		}
	}
	cout << maxid;
	return 0;
}