洛谷 P10131 Majority Opinion B 题解

题目简述

有 $N$ 头奶牛，每种奶牛都喜欢一种类型的干草 $h_i$，为了让所有奶牛都喜欢一种干草，Farmer John 会随意选中一个范围的奶牛组成一个小组，如果有一种干草是小组中超过一半的奶牛喜欢的，那么这个小组所有牛都会喜欢这种干草。求哪些类型的干草有可能变为同时受到所有奶牛的喜爱。

题目解法

一到有点思维含量的题目，但是可以通过观察得到结论。

这里我们从 $3$ 头奶牛组成的小来看，不难发现，只要有 $2$ 头奶牛喜好相同，这 $3$ 头奶牛就可以都喜欢一种干草了，此时我们再一次多框入少于之前小组数量的奶牛组成新的小组多次，就可以使一种干草覆盖整个序列。

因此，本题就是找任意连续三头奶牛 $\ge 2$ 头喜欢一种干草的所有情况。

如果没有任意连续三头牛中两个有相同的喜好，那么无论怎么进行分组都不可能符合超过半数的奶牛有相同喜好。例如：1 2 3 1 ，虽然有两个 $1$，单要想将这两个 $1$ 同时包括在内就必须将四头奶牛分成一组，这样就没有超过半数，无法改变其它奶牛的喜好。

最后，我用的桶来解决的，因为可以去重，排序。

参考代码

#include<bits/stdc++.h>
using namespace std;
int a[200010],book[200010],ans[110],t,n;
int main()
{
    cin>>t;
    while(t--){
        int f=1,cnt=0;
        memset(book,0,sizeof(book));
        memset(a,-1,sizeof(a));
        cin>>n;
        for(int i=2;i<=n+1;i++){
            cin>>a[i];
            if(a[i]==a[i-1]) book[a[i]]++;
            if(a[i]==a[i-2]) book[a[i]]++;
            if(a[i-1]==a[i-2]) book[a[i+2]]++;
        }
        for(int i=1;i<=200000;i++) 
            if(book[i]>0) f=0,ans[++cnt]=i;
        if(f==1) ans[++cnt]=-1;
        for(int i=1;i<=cnt;i++){
            if(i==cnt) cout<<ans[i];
            else cout<<ans[i]<<' ';
        }
        if(t>=1) cout<<endl;
    }
}

最后说一下，由于原题要求行末没有多余空格，所以我这里有处理这个问题的部分。