题目链接
题目大意:有一个长度为n的隐藏数组a,其中元素两两不同。你有一台机器可以最多询问n次,每次可以向这个机器输入k个数,这些数是下标(范围1到n)。而机器会返回这些数中第m大的数的值,以及他的下标。已知k,求m。
题解:其实解法在样例就有体现(后悔当时没有研究样例)。我们只需对前k+1个数查询k+1次,这样只会得到两个元素的值,一个是第m大的数,一个是第m+1大的数。第m大的数会出现(k + 1 - m)次, 第m + 1大的数会出现 m 次。所以我们只需要统计较大值出现的次数就是答案m了。
那为什么第m大的数会出现(k + 1 - m)次, 第m + 1大的数会出现 m 次?
答:当我们对前k+1个数询问k+1次时,实际上就是依次从k+1个数中去掉一个数进行查询。如果去掉的是前1 ~ m中的某一个数,那询问所得到的值则只会是第m + 1大的数。剩下的去掉m+1 ~ k + 1中的数,询问所得到的值必将会是第m大的数。
#include<bits/stdc++.h>
using namespace std;
map<int, int>mp;
int main()
{
int n, k, mx = 0;
cin>>n>>k;
for(int j = 1; j <= k + 1; ++j){
printf("?");
for(int i = 1; i <= k + 1; ++i){
if(i != j) printf(" %d", i);
}
printf("\n");
fflush(stdout);
int p, x;
scanf("%d%d", &p, &x);
mp[x]++;
mx = max(mx, x);
}
printf("! %d\n", mp[mx]);
}