본문 바로가기

TIL/Algorithm

[Algorithm] Hackerrank - Sock Merchant

반응형

문제:

John works at a clothing store. He has a large pile of socks that he must pair by color for sale. Given an array of integers representing the color of each sock, determine how many pairs of socks with matching colors there are.

For example, there are  socks with colors . There is one pair of color  and one of color . There are three odd socks left, one of each color. The number of pairs is .

Function Description

Complete the sockMerchant function in the editor below. It must return an integer representing the number of matching pairs of socks that are available.

sockMerchant has the following parameter(s):

  • n: the number of socks in the pile
  • ar: the colors of each sock

Input Format

The first line contains an integer , the number of socks represented in . 
The second line contains  space-separated integers describing the colors  of the socks in the pile.

 

Output Format

Return the total number of matching pairs of socks that John can sell.

Sample Input

9

10 20 20 10 10 30 50 10 20

Sample Output

3

Explanation

코드:

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the sockMerchant function below.
def sockMerchant(n, ar):
    from collections import defaultdict
    sum_ = 0
    a = defaultdict(lambda: 0)
    for i in ar:
        a[i] += 1
    for i in a:
        sum_ += int(a[i] / 2)
    print(sum_)
    return sum_

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    n = int(input())

    ar = list(map(int, input().rstrip().split()))

    result = sockMerchant(n, ar)

    fptr.write(str(result) + '\n')

    fptr.close()
반응형