题目戳我
样例过,时间全3ms不知道哪里错了
#include <stdio.h>
long long n, k, cnt = 0, sq = 1, num = 1;
int main() {
scanf("%lld%lld", &n, &k);
if (k % 2)
while (num <= n)
cnt += (num % k == 0 && num / k - k / 2 && n >= num / k + k / 2), sq++, num = sq * sq;
else
while (num <= n)
cnt += (num % k == 1 && num / k - k / 2 + 1 && n >= num / k + k / 2), sq++, num = sq * sq;
return printf("%lld\n", cnt) & 0;
}```