问题描述
n个小伙伴(编号从0到n-1)围坐一圈玩游戏。按照顺时针方向给n个位置编号,从0到n-1。
最初,第0号小伙伴在第0号位置,第1号小伙伴在第 1 号位置,……,依此类推。
游戏规则如下:每一轮第 0 号位置上的小伙伴顺时针走到第 m 号位置,第 1 号位置小伙伴走到第m+1号位置,……,依此类推,第n−m号位置上的小伙伴走到第0号位置,第n-m+1 号位置上的小伙伴走到第1号位置,……,第 n-1 号位置上的小伙伴顺时针走到第m-1号位置。
现在,一共进行了10的k次方轮,请问x号小伙伴最后走到了第几号位置。
输入格式
输入共1行,包含 4个整数n、m、k、x,每两个整数之间用一个空格隔开。
输出格式
输出共1行,包含 1个整数,表示10的k次方轮后x号小伙伴所在的位置编号。
样例输入
10 3 4 5
样例输出
5
数据规模和约定
1<n<100000
0<m<n,
0<=x<n
0<k<1000000000。
首先我们来分析一下题目:
根据题目的描述,大致的意思就是说n个学生绕圈,每次绕m个位置,重复这个操作10^k,第一个同学的编号为0,求第x个同学最后的位置
这道题的思路比较简单:
先记录下第0号同学的位置,然后得出执行10^k后0号同学的位置,再加上x同学的编号即是最终的结果
代码如下:
n, m, k, x = map(int, input().split())
position_stu0 = 0 # 记录0号学生的位置
position_stu0 = (position_stu0 + m) * 10 ** k % n
print((position_stu0 + x) % n)
然后就只拿到了可怜的70分
仔细思考一下,超时的原因也比较明显,就是10^k这个数在k值很大的时候,会造成大量的计算,这也就导致了我们运算的超时,这时候我们就该想到了降幂。
建议去看一下这位大佬的文章快速幂算法
大致就是说,我们对一个数进行降幂,3^20=(3^2)^10=((3^2)^2)^5……,但是实际上仅仅这样降幂是不会节省时间的,我们对这个数进行降幂的目的其实是对每一次运行结果进行一个求余,使它的数位变小,从而节省时间。这是因为在计算机运算的时候,数值越大,计算后的进位也会很大,运算时间相应的也会更长。根据这道题来看,我们的余数就是n的值了。不管多少个人,执行一次同人数的轮回,最终的结果都是一样的,这个道理懂得都懂好吧
来看优化后的代码:
def drop_power(n, num, power):
result = 1
while power:
if power & 1: # 和判断power是否为1是一样的
result = result * num % n # 幂为奇数的时候,分离出一次底数
num = num ** 2 % n # 幂为偶数,直接降幂
power >>= 1 # power右移一位,和power/2是一样的
return result
n, m, k, x = map(int, input().split())
position_stu0 = 0
drop_result = drop_power(n, 10, k)
position_stu0 = (position_stu0 + m) * drop_result % n
print((position_stu0 + x) % n)
ok也是成功拿下!