B. Range and Partition

原題連結

題解

細節

1.劃分陣列時要用原本的程式碼，而不是排序後的程式碼
2.更新劃分陣列時一些引數要更新

code

#include<bits/stdc++.h>
#define ll long long
using namespace std;

int a[200005],b[200005];

void solve()
{
    int n,k;
    cin>>n>>k;

    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        b[i]=a[i];
    }

    sort(a+1,a+1+n);

    int len=(n+k)/2+(n+k)%2;

    int x=a[1],y=a[len],ans=y-x;
    for(int r=len+1;r<=n;r++)
    {
        int l=r-len+1;

        if(a[r]-a[l]<ans)
        {
            ans=a[r]-a[l];
            x=a[l];
            y=a[r];
        }
    }

    cout<<x<<' '<<y<<'\n';
    int l=1,cnt=0,sum=0;
    for(int i=1;i<=n;i++)
    {
        if(b[i]<=y&&b[i]>=x) cnt++;
        if(2*cnt>i-l+1&&sum!=k-1)
        {
            sum++;
            cout<<l<<' '<<i<<'\n';
            l=i+1;
            cnt=0;
        }
    }

    cout<<l<<' '<<n<<'\n';
    cout<<'\n';
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t=1;
    cin>>t;
    while(t--) solve();
    return 0;
}