leetpycode/easy/kth_distinct_string_in_an_array.py

10 lines
414 B
Python

# https://leetcode.com/problems/kth-distinct-string-in-an-array
from typing import List
class Solution:
def kthDistinct(self, arr: List[str], k: int) -> str:
cnt = dict()
for i in range(len(arr)):
cnt[arr[i]] = len(arr) if arr[i] in cnt else i
ret = sorted(cnt.values())[k - 1] if 0 <= k - 1 < len(cnt) else -1
return "" if ret < 0 or ret >= len(arr) else arr[ret]