#include "kthnumber.h" int find_K(int N, int M, int subtask) { // TODO: Implement your solution // This function will be called T times per test case // Be careful if using global variables (you may want to reset them each time this function is called) // this code asks a question with the first M indices std::vector I; for (int i = 0; i < M; i++) { I.push_back(i); } int result = ask_host(I); return 1; // we guess K = 1 }