-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathB_Battle_Cows.cpp
68 lines (56 loc) · 1.18 KB
/
B_Battle_Cows.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/*
RunAt - green
*/
#include <iostream>
#include <set>
#include <climits>
#include <string>
#include <numeric>
#include <cmath>
#include <map>
#include <algorithm>
#include <vector>
using namespace std;
typedef int long long ll;
#define p(a) cout << a << "\n";
#define p2(a,b) cout << a << " " << b << "\n";
#define YES cout << "YES" << "\n"
#define NO cout << "NO" << "\n"
const int N = 1e5+5;
const int MOD = 1e9+7;
vector<ll> getWins(vector<ll> v) {
int n = v.size();
vector<ll> wins(n);
for (int i=1,j=0; i<n; i++) {
if (v[j] < v[i]) j = i;
wins[j]++;
}
return wins;
}
void solve() {
ll n, k;
cin >> n >> k;
vector<long long> vec(n);
for (auto &e: vec) cin >> e;
auto wins = getWins(vec);
ll ans = wins[k-1];
swap(vec[k-1], vec[0]);
wins = getWins(vec);
ans = max(ans, wins[0]);
swap(vec[k-1], vec[0]);
ll firstGreater = find_if(vec.begin(), vec.end(),
[&] (int e) {return e > vec[k-1];}
) - vec.begin();
swap(vec[k-1], vec[firstGreater]);
wins = getWins(vec);
ans = max(ans, wins[firstGreater]);
p(ans);
}
int main() {
cin.sync_with_stdio(0);
cin.tie(0);
int tests = 1;
cin >> tests;
while (tests--) solve();
return 0;
}