탑코더 알고리즘 정복 - 4. 재미있는 수학

class InterestingDigits:
	def digits(self, base):

# Case# 0
base = 10
returns = [3, 9]

# Case# 1
base = 3
returns = [2]

# Case# 2
base = 9
returns = [2, 4, 8]

# Case# 3
base = 26
returns = [5, 25]

# Case# 4
base = 30
returns = [29]

class InterestingDigits:
    def digits(self, base):
        ans = []

        for n in range(2, base):
            ok = True
            for k1 in range(base):
                for k2 in range(base):
                    for k3 in range(base):
                        if (k1 + k2 * base + k3 * base * base) % n == 0 and (k1 + k2 + k3) % n != 0:
                            # n의 배수 숫자로 각 숫자의 합이 n의 배수가 아니면
                            # n을 제외합니다.
                            ok = False
                            break
                    if not ok:
                        break
                if not ok:
                    break
            if ok:
                ans.append(n)

        return ans

class InterestingDigits:
    def digits(self, base):
        ans = []

        for i in range(2, base):
            if ((base - 1) % i) == 0:
                ans.append(i)

        return ans