codeforces 888 div.3 C题

 题目链接  Problem - C - Codeforces

 代码

#include<iostream>
#include<algorithm>
using namespace std;

int x[200001];

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t, n, k, i, sum, kk, j;
    cin >> t;
    while (t--)
    {
        cin >> n >> k;
        for (i = 0, kk = 0, sum = 0; i < n; i++)
        {
            cin >> x[i];
            if (kk == 0 && x[i] == x[0])sum++;
            if (kk == 0 && sum == k)kk = 1, j = i;
        }
        if (kk == 1)
        {
            if (x[n - 1] == x[0])
                cout << "YES\n";
            else
            {
                for (i = j + 1, sum = 0; i < n; i++)
                    if (x[i] == x[n - 1])sum++;
                if (sum >= k)cout << "YES\n";
                else cout << "NO\n";
            }
        }
        else cout << "NO\n";
    }
    return 0;
}