AT_ttpc2023_c Yet Another Simple Math Problem

题目描述

给定一个整数 $N$。请你求满足以下两个条件的正整数对 $(a, b)$ 的个数。 - $1 \leq a, b \leq N$ - 存在一组正整数 $(x, y)$,使得 $x + y^2 = a$ 且 $x^2 + y = b$ 同时成立 有 $T$ 组测试数据,请你分别作答。

输入格式

输入通过标准输入给出,格式如下: > $T$ $\mathrm{case}_1$ $\mathrm{case}_2$ $\vdots$ $\mathrm{case}_T$ 每组测试数据 $\mathrm{case}_i\ (1 \leq i \leq T)$ 的格式为: > $N$

输出格式

输出 $T$ 行。对于第 $i$ 个测试数据,请在第 $i$ 行输出对应的答案。

说明/提示

### 样例解释 1 对于第 $1$ 个测试数据,满足条件的 $(a, b)$ 共有 $4$ 组,分别为 $(a, b) = (2, 2), (3, 5), (5, 3), (6, 6)$。 例如,对于 $(a, b) = (3, 5)$,取 $(x, y) = (2, 1)$,此时 $x + y^2 = 3 = a$,$x^2 + y = 5 = b$,条件成立。 对于第 $2$ 个测试数据,满足条件的 $(a, b)$ 不存在。 ### 数据范围 - $1 \leq T \leq 10^5$ - $1 \leq N \leq 10^{18}$ - 所有输入均为整数。 由 ChatGPT 5 翻译