题目:
监狱有n 个房间,每个房间关押一个犯人,有 m 种宗教,每个犯人会信仰其中一种。如果相邻房间的犯人的宗教相同,就可能发生越狱,求有多少种状态可能发生越狱。答案对100,003 取模。
思路:
1.整体思路:
可能越狱的状态数量=所有状态数量-无法越狱状态数量(n个数字任意两个相邻的数字都不相同)=m^n-m*(m-1)^(n-1)
2.计算a^b:
使用二分法:若b为奇数,a^b=a^(b/2)*a^(b/2)*a; 若b为偶数,a^b=a^(b/2)*a^(b/2)。
3.取余时要注意:
由于数据过大,我们最后计算出的ans其实是(m^n)%100003-(m*(m-1)^(n-1))%100003;而这会出现一个问题,那就是两个余数相减很有可能是个负数。(比如:4%2-1%2=-1,它不等于(4-1)%2!)
代码展示: