nowcoder 蚂蚁

题目

n只蚂蚁从正n边形的n个定点沿着边移动，速度是相同的，问它们碰头的概率是多少？
给定一个正整数n，请返回一个数组，其中两个元素分别为结果的分子和分母，请化为最简分数。
测试样例：
3
返回：[3,4]

思路

$\frac{2^n-2}{2^n}$

代码

class Ants:
    def gcd(self, x, y):
        while y:
            tmp = y
            y = x % y
            x = tmp
        return x
    def collision(self, n):
        # write code here
        num1 = 2 ** n
        num2 = num1 - 2
        gcd = self.gcd(num1, num2)
        return [num2 // gcd, num1 // gcd]