Chương trình đếm số palindrome độ dài k từ ký tự chuỗi trong Python

Cho chuỗi s và số k. Đếm số palindrome độ dài k có thể tạo từ các ký tự trong s (có thể dùng lại ký tự).

Ví dụ

Đầu vào: s = "xy", k = 4.
Đầu ra: 4 (palindrome: "xxxx", "yyyy", "xyyx", "yxxy").

Thuật toán

1. n = k // 2.
2. x = số ký tự duy nhất trong s.
3. Kết quả = x^(n + k%2).

Mã nguồn Python

class Solution:
    def solve(self, s, k):
        n = k // 2
        return len(set(s)) ** (n + k % 2)

s = "xy"
k = 4
ob = Solution()
print(ob.solve(s, k))

Đầu vào / Đầu ra

Input: "xy",4
Output: 4