题目背景
时/空限制:1s / 64MB
题目描述
农夫约翰有一项重要的任务——弄清楚要为他的奶牛们购买什么类型的干草。
农夫约翰的 N 头奶牛编号为 1 到 N ,每头奶牛喜欢恰好一种类型的干草 hi 。
他希望他的所有奶牛都喜欢同一种干草。
为了实现这一目标,农夫约翰可以主持焦点小组访谈。
每一次焦点小组访谈,约翰都可以自由选择任意多个连续编号的奶牛构成访谈小组,共同参加访谈。
如果有一种干草是小组中超过一半的奶牛喜欢的,则此次焦点小组访谈结束后,组内所有奶牛最终都会喜欢这种干草。
如果不存在这样的干草,那么奶牛们就不会改变她们喜欢的干草类型。
例如,在由 16 头奶牛组成的焦点小组访谈中,需要有其中 9 头或更多的奶牛具有相同的干草喜好,才能使其余奶牛改变其喜好以与之一致。
农夫约翰想知道哪些类型的干草有可能变为同时受到所有奶牛的喜爱。
他一次只能主持一个焦点小组访谈,但为了使所有奶牛都喜欢同一类型的干草,他可以根据需要任意多次地主持焦点小组访谈。
输入格式
输入的第一行包含一个整数 T ,为独立的测试用例的数量。
每一个测试用例的第一行包含 N 。
第二行包含 N 个整数,为奶牛们喜爱的干草类型 hi 。
输出格式
输出 T 行,对于每个测试用例输出一行。
如果可能使所有奶牛同时喜欢同一种干草,则以升序输出所有可能的此类干草的类型,否则输出 -1。
在同一行内输出一列整数时,相邻的数用空格分隔,并确保行末没有多余空格。
输入输出样例
输入 #1复制
5
5
1 2 2 2 3
6
1 2 3 1 2 3
6
1 1 1 2 2 2
3
3 2 3
2
2 1
输出 #1复制
2
-1
1 2
3
-1
说明/提示
数据范围
1≤T≤10 ,
2≤N≤10^5 ,
1≤hi≤N ,
输入保证同一测试点内所有测试用例的 N 之和不超过 2⋅10^5 。
样例解释
在输入样例中,有 5 个测试用例。
在第一个测试用例中,仅可能使所有奶牛喜欢种类 2 。约翰可以通过主持一次所有奶牛的焦点小组访谈达到这一目的。
在第二个测试用例中,可以证明没有奶牛会改变她们喜爱的干草种类。
在第三个测试用例中,有可能使所有奶牛喜欢种类 1 ,可以通过主持三次焦点小组访谈达到这一目的——首先使奶牛 1 到 4 进行一次焦点小组访谈,随后使奶牛 1 到 5 进行一次焦点小组访谈,随后使奶牛 1 到 6 进行一次焦点小组访谈。以类似的逻辑,依次操作奶牛 3 到 6 ,随后是奶牛 2 到 6 ,随后是奶牛 1 到 6 ,我们可以使所有奶牛喜欢种类 2 。
在第四个测试用例中,有可能使所有奶牛喜欢种类 3 ,可以通过主持一次所有奶牛的焦点小组访谈达到这一目的。
在第五个测试用例中,可以证明没有奶牛会改变她们喜爱的干草种类。
请求帮助!!!(必关)
代码:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> possibleHayTypes(vector<int>& hayTypes, int N) {
int maxCount = 0;
int currentType = -1;
int currentCount = 0;
for (int i = 0; i < N; ++i) {
if (hayTypes[i] == currentType) {
++currentCount;
} else {
if (currentCount > N / 2) {
maxCount = max(maxCount, currentCount);
}
currentType = hayTypes[i];
currentCount = 1;
}
}
if (currentCount > N / 2) {
maxCount = max(maxCount, currentCount);
}
vector<int> possibleTypes;
if (maxCount > N / 2) {
for (int i = 1; i <= N; ++i) {
if (count(hayTypes.begin(), hayTypes.end(), i) > N / 2) {
possibleTypes.push_back(i);
}
}
sort(possibleTypes.begin(), possibleTypes.end());
} else {
possibleTypes.push_back(-1);
}
return possibleTypes;
}
int main() {
int T;
cin >> T;
while (T--) {
int N;
cin >> N;
for (int i = 1; i <= N; i++) {
vector<int> hayTypes(N);
for (int i = 0; i < N; ++i) {
cin >> hayTypes[i];
}
vector<int> result = possibleHayTypes(hayTypes, N);
if (result.size() == 1 && result[0] == -1) {
cout << -1 << endl;
} else {
for (size_t i = 0; i < result.size(); ++i) {
cout << result[i];
if (i < result.size() - 1) cout << " ";
}
cout << endl;
}
}
}
return 0;
}