2021牛客寒假算法基础集训营2 D.牛牛与整除分块（数论）

2021牛客寒假算法基础集训营2 D.牛牛与整除分块（数论）

题目链接：https://ac.nowcoder.com/acm/contest/9982/D（付费比赛）
题面：
题解：考虑集合 $S$ 中的元素组成，首先集合 $S$ 中一定存在元素 { 1 , 2 , ⋯   , ⌊ N ⌋ − 1 } \{1,2,\cdots,\lfloor\sqrt{N}\rfloor-1\} {1,2,⋯,⌊N ​⌋−1} 。这是因为：

$\forall i\in [\lfloor \frac{N}{2}\rfloor +1,N]$，有 $\lfloor \frac{N}{i}\rfloor =1$ ，

$\forall i\in [\lfloor \frac{N}{3}\rfloor +1,\lfloor \frac{N}{2}\rfloor ]$，有 $\lfloor \frac{N}{i}\rfloor =2$ ，

$\cdots \cdots$

$\forall i\in [\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor +1,\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor -1}\rfloor ]$，有 $\lfloor \frac{N}{i}\rfloor =\lfloor \sqrt{N}\rfloor -1$ 。

因此，集合 $S$ 一定包含 $1$ 到 $\lfloor \sqrt{N}\rfloor -1$ 的自然数。显然，对于剩下的区间 ：
$\forall i,j\in [1,\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor ](i\ne j)$，有 $\lfloor \frac{N}{i}\rfloor \ne \lfloor \frac{N}{j}\rfloor$ ，且对于 $\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor$ ，有 $\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor =\lfloor \sqrt{N}\rfloor +1$ 或 $\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor =\lfloor \sqrt{N}\rfloor$ 。那么，这个集合的元素即为 { 1 , 2 , 3 , ⋯   , ⌊ N 3 ⌋ , ⌊ N 2 ⌋ , N } \{1,2,3,\cdots,\lfloor\frac{N}{3}\rfloor,\lfloor\frac{N}{2}\rfloor,N\} {1,2,3,⋯,⌊3N​⌋,⌊2N​⌋,N} ，当且仅当$\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor =\lfloor \sqrt{N}\rfloor$ 时，这个集合会减少一个重复的元素 $\lfloor \sqrt{N}\rfloor$ ，或者说是$\lfloor \frac{N}{\lfloor \sqrt{N}\rfloor }\rfloor$ 。
不难看出，这个集合的左右部分是有机关联的，在每个对称的位置 $(x,y)$ 上，一定有 $\lfloor \frac{N}{x}\rfloor =y$ ，且这个集合的元素数目总量总是 $2\lfloor \sqrt{N}\rfloor$ 或 $2\lfloor \sqrt{N}\rfloor -1$ 。那么对于任意一个数 $\lfloor \frac{N}{x}\rfloor$ ，我们总可以根据它是否大于 $\lfloor \sqrt{N}\rfloor$ 来分别判别它的位置。如果大于 $\lfloor \sqrt{N}\rfloor$ ，那么数 $x$ 就是它所在的降序序列的位置，反之，可以通过观察这个元素的对称位置来间接查找它的位置。
下面是AC代码。

#include<bits/stdc++.h>
using namespace std;
int main(){
    int T,N,x;
    scanf("%d",&T);
    while(T--){
        scanf("%d %d",&N,&x);
        int z=sqrt(N);
        int ans=2*z-N/x+1;
        if(z==N/z) --ans;
        if(x<=z) printf("%d\n",x);
        else printf("%d\n",ans);
    }
}