airwrk codding test (21/12/2022)

problem1.py

from typing import List


def good_pair(nums: List[int]) -> int:
    good_pair_count = 0
    for i in range(len(nums)):
        for j in range(i+1, len(nums)):
            if nums[i] == nums[j] and i < j:
                good_pair_count += 1
    return good_pair_count


nums = [1, 2, 3]
ans = good_pair(nums)
print(ans)

problem2.py

from collections import Counter
from typing import List


def find_single(nums: List[int]) -> int:
    freq = Counter(nums)
    for key in freq:
        if freq[key] == 1:
            return key
    return None


nums = [4, 1, 2, 1, 2]
ans = find_single(nums)
print(ans)

problem3.py

def truncate_sentence_to_k(s: str, k: int) -> str:
    splitted_str = s.split(" ")
    return " ".join(splitted_str[:k])


s, k = "What is the solution to this problem", 4
ans = truncate_sentence_to_k(s, k)
print(ans)