温澈滢的狗狗(二分+尺取)

题目链接：https://ac.nowcoder.com/acm/contest/9984/D

题目大意

n只狗狗颜色是ai，颜色不同的狗狗之间有亲密关系，亲密度是下标差。将所有亲密关系按照先亲密度，后第一只狗狗编号，后第二支狗狗编号排序，求第k个亲密关系的两只狗狗的编号。（1<=1e5<=n,1<=ai<=n,1<=k<=n*(n-1)/2）

思路

二分第k对关系的亲密度，cal(x)表示亲密度小于等于x的亲密关系数，反着求，就是下标差小于等于x的所有关系数 - 同色的下标差小于等于x的关系数。

所有关系数是x * (n-x)+(x-1)*x/2。比如[1, 2, 3, 4, 5, 6]（下标），x=3,如果第一只狗是1的话，那么第二只狗可以是2,3,4,总共x种，第一只狗是2的话，第二只狗最大可以是5，也有x种，但是当第一只狗是4的话，第二只狗最大也只能6，只有x-1种，后面是x-2, x-1 ,,, 1。所以有n-x个x加上(1,2...,x-1)的求和。同色关系数可以用将同色下标存vector，然后用尺取法求解。

如果cal(x)<=k，那么x要变大，l=x+1;否则x要变小，r=mid-1。然后可以明确第k对关系亲密度，当k-cal(x)==0时，说明亲密度是x，并且是最后一对，那么可以倒着求第一个关系度为x的那对。当k-cal(x)>0时，说明他是亲密度为x+1的第k-cal(x)对，那么可以顺着找。

注意特判cal(n-1)<k的情况，因为不存在。然后记得开ll，因为k会爆int

ac代码

#include<bits/stdc++.h>
using namespace std;
mt19937_64 rng(time(0));
#define io cin.tie(0);ios::sync_with_stdio(false);
#define ok(x, y) x >= 1 && x <= n && y >= 1 && y <= m
#define debug(x) cout<<#x<<"="<<x<<endl
#define lowbit(x) x&(-x)
#define pii pair<int,int>
#define mk make_pair
#define ll long long
#define ull unsigned long long
#define rs p<<1|1
#define ls p<<1
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 1e18;
inline ll read(){
    ll p=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){p=(p<<1)+(p<<3)+(c^48),c=getchar();}
    return f*p;
}
void print(__int128 x){
    if(x<0) {putchar('-'); x=-x;}
    if (x>9) print(x/10);
    putchar('0'+x%10);
}

int a[maxn];
vector<int> v[maxn];
ll n, k;
ll cal(ll x){
    ll cc = 0;
    for(int i = 1; i <= n; i ++){
        // 尺取法求解
        int r = 0;
        for(int l = 0; l < v[i].size(); l ++){
            while(r < v[i].size() && v[i][r] - v[i][l] <= x) r ++;
            cc += r - l - 1;
        }
    }
    return x * (n - x) + (x - 1) * x / 2 - cc; //亲密度小于等于x的关系数
}
void solve(){
    cin >> n >> k;
    for(int i = 1; i <= n; i ++){
        cin >> a[i];
        v[a[i]].push_back(i); //存同颜色下标
    }
    if(cal(n - 1) < k){ //特判第k对关系是否存在
        cout << -1 << endl;
        return;
    }
    int l = 0, r = n - 1, ans = 0;
    while(l <= r){ //二分亲密度
        int mid = l + r >> 1;
        if(cal(mid) <= k) l = mid + 1, ans = mid;
        else r = mid - 1;
    }
    ll res = k - cal(ans);
    if(res == 0){ 
        for(int i = n - ans; i >= 1; i --){ //倒着找第一个
            int j = i + ans; //保证相差ans
            if(a[i] != a[j]) {  //保证异色
                cout << i << " " << j << endl;
                return;
            }
        }
    }
    ans ++;
    for(int i = 1; i + ans <= n; i ++){
        int j = i + ans;
        if(a[i] == a[j]) continue;
        if(-- res == 0){ //顺着第k-cal(ans)个
            cout << i << " " << j << endl;
            return;
        }
    }

}
int main(){
    // freopen("1.in", "r", stdin);
    // freopen("std.out", "w", stdout);
    io;
    solve();
    return 0;
}