找区间

题面描述：
给定 n 个数 a1, …, an， 高老师想了解 al, …, ar 中有多少对相邻元素值相同。 高老
师把这个数目定义为区间[l, r]的价值，用 v[l, r]表示。例如 1,1,1,2,2 这五个数所
组成的区间的价值为 3。
现在高老师想知道在所有的 v[l, r] (1≤l≤r≤n)中， 第 k 小的值是多少。 但高老师要
和女朋友出去玩，于是他把这个问题甩给了你，请你帮他解决一下。
输入数据：
第一行有一个整数 t(1≤t≤30)， 表示有 t 组数据。
对于每组数据：
第一行有两个整数 n, k (1≤n≤2000,1≤k≤n(n+1)/2)；
第二行有 n 个整数 a1, … , an (1≤ai≤109)。
输出数据：
对于每组数据：
输出一个整数，表示第 k 小的值。
样例输入：
2
4 7
1 1 2 3
3 5
100 100 100
样例输出：
0 1
程序源码：

#include<cstdio>
#include<cmath>
#include<cstring>
#include<map>
#include<algorithm>
using namespace std;
typedef long long ll;
ll a[2001];
ll t, n, m, temp[2001];
ll vis[2001];
ll check(ll mid)//尺取求比mid小的区间个数
{
	int i, j;
	ll sum = 0;
	ll num = 0;
	memset(vis, 0, sizeof(vis));
	for (i = 0, j = 0; i<n; i++)
	{
		for (; j<n&&sum + vis[a[j]] <= mid; j++)
		{
			sum += vis[a[j]];
			vis[a[j]]++;
		}
		num += j - i;//尺取思想核心
		vis[a[i]]--;
		sum -= vis[a[i]];
	}
	return num >= m;
}
int main()
{
	ll ans;
	scanf_s("%lld", &t);
	while (t--)
	{
		scanf_s("%lld%lld", &n, &m);
		for (int i = 0; i<n; i++)
		{
			scanf_s("%lld", &a[i]);
			temp[i] = a[i];
		}
		int cnt;
		//sort(temp, temp + n);//离散化
		cnt = unique(temp, temp + n) - temp;
		for (int i = 0; i < n; ++i)a[i] = lower_bound(temp, temp + cnt, a[i]) - temp;
		ll left = 0, right = ((ll)n*((ll)n - 1)) / 2;
		while (left <= right)
		{
			ll mid = (left + right) / 2;
			if (check(mid))//如果小于mid的区间个数比m多的话，说明值还不够大
			{
				ans = mid;
				right = mid - 1;
			}
			else {
				left = mid + 1;
			}
		}
		printf("%lld\n", ans);
	}
	system("pause");
	return 0;
}