[Interview Preparation Kit] Warm-up-1

이희진·2022년 11월 23일
0

Problem

There is a large pile of socks that must be paired by color. Given an array of integers representing the color of each sock, determine how many pairs of socks with matching colors there are.

Example
n = 7
ar = [1,2,1,2,1,3,2]

There is one pair of color 1 and one of color 2. There are three odd socks left, one of each color. The number of pairs is 2.

Function Description
Complete the sockMerchant function in the editor below.
sockMerchant has the following parameter(s):

int n: the number of socks in the pile
int ar[n]: the colors of each sock

Returns
int: the number of pairs

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

My solution

#!/bin/python3

import math
import os
import random
import re
import sys

def sockMerchant(n, ar):
    not_pair = []
    pair = 0
    for num in ar:
        if num in not_pair:
            not_pair.remove(num)
            pair += 1
        else:
            not_pair.append(num)
    return pair
            

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

    n = int(input().strip())

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

    result = sockMerchant(n, ar)

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

    fptr.close()

0개의 댓글