题意:一个长度为 n 全为正整数的数组,对于所有点,有边(l,r)(1 ≤ l < r ≤ n)为min(al……ar)。可以修改 k 个数的值,求出图可能的最大直径。
思路:设整个数组最小值为 m,对于任意的边(l,r) 如果 m 在 l 和 r 之间那么最短路径就是 m,否则只要通过 l -> m -> r 这个路径,那么最大的最短路径就是 2m,也就是图的直径最大为 2m。对于一条边(ai,ai + 1),其他任意边(ai,r)(ai+1 < r ≤ n),边(l,ai+1)(1 ≤ l < ai),一定小于等于边(ai,ai + 1),所以最大的边为边(ai,ai + 1)(1 ≤ i < n)的最大值,那么图的最大直径为min(2m,所有边(ai, ai + 1)的最大值),然后通过二分求出修改 k 个数图的最大直径 x。对于最大直径 x 我们必须保证数组中所有数的两倍都大于等于 x,而且至少有一条边(ai,ai+ 1)大于等于 x。
代码
#include<bits/stdc++.h>
#define pb push_back
#define all(x) x.begin(), x.end()
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 2e5 + 10, P = 1e9 + 7, mod = 998244353;
int n, k;
vector<int> b;
bool check(int x){
int cnt = 0;
for(int i = 1; i <= n; i++)
if(b[i] * 2 < x) cnt++, b[i] = 1e9;
bool f1 = 0, f2 = 0;
for(int i = 1; i <= n; i++){
if(b[i] >= x) f1 = 1;
if(i < n && min(b[i], b[i + 1]) >= x) f2 = 1;
}
cnt += !f1 + !f2;
return cnt <= k;
}
void solve(){
cin >> n >> k;
vector<int>a(n + 1);
for(int i = 1; i <= n; i++) cin >> a[i];
int l = 0, r = 1e9;
while(l < r){
int mid = l + r + 1 >> 1;
b = a;
if(check(mid)) l = mid;
else r = mid - 1;
}
cout << l << endl;
}
int main(){
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int tt;
cin >> tt;
while(tt--) {
solve();
}
return 0;
}