有 N个方块，索引为 0到 (N-1)排列在一条线上。Snuke将通过以下程序来标记每个方格。 标记方格 0. 重复以下步骤N-1次。 初始化一个变量 x=(A+D)modN，其中 A是上次标记的方块的索引。 当方格 x被标记，重复将 替换为 (x+1)modN。 标记方格 x. 找出Snuke第K次标记的广场的索引。 给出 T个测试案例，找出每个案例的答案。 N,K,D最大为10^9 给出一个时间复杂度小于O(logk)的解法，并给出代码

由于要求时间复杂度小于 O(logk)，可以使用快速冥想算法： 令x=0，令A=1，令B=D 对K次迭代： 如果 K 为奇数，则 x=(Ax+B) mod N 如果 K 为偶数，则 A=(A*A) mod N K=K/2 最后x的值即为第K次标记的方格的索引。 以上方法的时间复杂度为 O(logK)。 以C语言为例，代码如下： int x=0, A=1, B=D, K; while(K)
{
if (K & 1)
x = (A * x + B) mod N;
A = (A * A) mod N;
K >>= 1;
}
printf("%d\n", x);

相关问题

有 N个方块，索引为 0到 (N-1)排列在一条线上。Snuke将通过以下程序来标记每个方格。 标记方格 0. 重复以下步骤i - iii (N-1)次。 初始化一个变量 x有 (A+D)modN，其中 A是上次标记的方块的索引。 当方格 x被标记，重复将 替换为 (x+1)modN。 标记方格 x. 找出Snuke第K次标记的广场的索引。 给出 T个测试案例，找出每个案例的答案。 N,K,D最大均为10^9，给出一个时间复杂度不超时的算法

由于N，K，D的最大值都为10^9，推荐使用高效的循环移位算法来解决这个问题。首先令x=A，令r=0，令m=1，令t=1，令y=0，令z=K；令i=0，令j=1，令k=0；然后，当i<N时，执行以下操作：r=r+m*t，m=m*D，t=t*D，x=x+y，y=y+z，i=i+1，j=j+1，k=k+1，当k=K时，输出索引i。最后，输出索引i即可得到Snuke第K次标记的方格的索引。