题目:镜面反射
题目:镜面反射
有一个特殊的正方形房间,每面墙上都有一面镜子。除西南角以外,每个角落都放有一个接受器,编号为 0, 1,以及 2。
正方形房间的墙壁长度为 p,一束激光从西南角射出,首先会与东墙相遇,入射点到接收器 0 的距离为 q 。
返回光线最先遇到的接收器的编号(保证光线最终会遇到一个接收器)
示例
输入: p = 2, q = 1
输出: 2
解释: 这条光线在第一次被反射回左边的墙时就遇到了接收器 2
题解
我们假设光不会反射,直接直行,如下图所示,由此我们可以得到以下结论
- 如果
q % 2 == 0
, 反射点为0
- 如果
p % 2 == 0
, 反射点为2
- 如果
q % 2 == 1
, 反射点为1
或2
- 如果
p % 2 == 1
, 反射点为0
或1
当 p
和 q
不同时为偶数时,我们可以简化结论
- 如果
q % 2 == 0
, 反射点为0
- 如果
p % 2 == 0
, 反射点为2
- 剩下情况,反射点为
1
func mirrorReflection(p int, q int) int {
for q % 2 == 0 && p % 2 == 0 {
p /= 2
q /= 2
}
if p % 2 == 0 {
return 2
}
if q % 2 == 0 {
return 0
}
return 1
}
有疑问加站长微信联系(非本文作者)