LeetCode 858 镜面反射
题目
有一个特殊的正方形房间,每面墙上都有一面镜子。除西南角以外,每个角落都放有一个接受器,编号为 0, 1,以及 2。
正方形房间的墙壁长度为 p,一束激光从西南角射出,首先会与东墙相遇,入射点到接收器 0 的距离为 q 。
返回光线最先遇到的接收器的编号(保证光线最终会遇到一个接收器)。
思路
本题若单纯依靠镜面反射的思想非常复杂难以解决问题,必须将问题转化。
1.我们假设东西两面墙无限向北拉长,将北墙暂时忽略,使得光线只有向上的反射。
2.当向上的纵距离第一次为p的整数倍时,可认为光线到达其中一个接收器。
3.通过观察画图可以发现,这个纵向距离是p和q的最小公倍数,设为s。
4.如何去判断光线遇到哪个接收器?分两步
a.首先确定是光线到达南墙还是北墙(判断s/p奇偶);
b.若是南墙(s/p为偶)直接返回接收器0;若是北墙(s/p为奇)还需确定是接受器1还是2(判断s/q奇偶);
代码
class Solution {
public:
int mirrorReflection(int p, int q) {
int s=lcm(p,q);//最小公倍数等于纵向距离
if((s/p)%2==0){//若南北次数为偶,则为南墙,0
return 0;
}
else{//为奇北墙
if((s/q%2)==1)return 1;//若东西次数为奇,则东墙,1
else return 2;//为偶西墙,2
}
}
int gcd(int a, int b){//最大公约数
return a % b ? gcd(b, a % b) : b;
}
int lcm(int a, int b){//最小公倍数
return a * b / gcd(a, b);
}
};