순서쌍이란 두 개의 숫자를 순서를 정하여 짝지어 나타낸 쌍으로 (a, b)로 표기합니다. 자연수 n이 매개변수로 주어질 때 두 숫자의 곱이 n인 자연수 순서쌍의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ n ≤ 1,000,000

풀이 1

def solution(n):
    answer = 0
    for i in range(1, n+1):
        j = n % i
        if j == 0:
            answer += 1
        else:
            continue
    return answer

• 답은 맞았는데, 몇몇 케이스에서 시간이 꽤 걸림 테스트 1 〉 통과 (0.00ms, 10.1MB) 테스트 2 〉 통과 (0.00ms, 10.1MB) 테스트 3 〉 통과 (7.78ms, 10.1MB) 테스트 4 〉 통과 (0.00ms, 10.3MB) 테스트 5 〉 통과 (5.75ms, 10.2MB) 테스트 6 〉 통과 (59.47ms, 10.1MB) 테스트 7 〉 통과 (0.00ms, 10.2MB) 테스트 8 〉 통과 (0.01ms, 10.1MB) 테스트 9 〉 통과 (62.75ms, 10.2MB) 테스트 10 〉 통과 (83.92ms, 10.2MB)

• idea 제곱수를 제외하면 같은 원소를 가진 순서쌍이 2개씩이니까, 연산량을 거의 절반으로 줄여볼 수 있지 않을까?

풀이 2

• idea 제곱근으로 이루어진 순서쌍은 1개, 그 외의 순서쌍은 2개씩 있다는 점을 이용했다

  def solution(n):
    answer = 0
    for i in range(1, int(n**(1/2))+1):
        if i*i == n:
            answer += 1
        elif n % i == 0:
            answer += 2
        else:
            continue
    return answer

• 풀이 1에 비하면 시간이 확연히 줄었다 테스트 1 〉 통과 (0.00ms, 10.1MB) 테스트 2 〉 통과 (0.01ms, 10.2MB) 테스트 3 〉 통과 (0.04ms, 10.2MB) 테스트 4 〉 통과 (0.02ms, 10.3MB) 테스트 5 〉 통과 (0.04ms, 10.1MB) 테스트 6 〉 통과 (0.10ms, 10.1MB) 테스트 7 〉 통과 (0.01ms, 10.3MB) 테스트 8 〉 통과 (0.02ms, 10.2MB) 테스트 9 〉 통과 (0.10ms, 10.1MB) 테스트 10 〉 통과 (0.10ms, 10.1MB)

• 하나 더 깨달은 점은, 동일한 변수를 게속 쓰는 게 아니면서, 코드 가독성에 무리가 없다면 변수를 따로 만들지 않고 바로 사용하는 편이 더 빠르다. 근데 코테볼 때는 속도가 중요하니까 그렇게 짜겠는데, 실제로는 0.1ms 더걸려도 변수 따로 선언해주는 게 좋지 않을까?

• (수정) 다시 보니까, n%i 조건을 먼저 쓰고, elif i*i ==n 을 쓰는 게 더 빠를 것 같다. 제곱근은 어차피 하나 밖에 없는 데 굳이 먼저 훑고 지나가게 만들 필요가 없다고 생각된다.

다른 사람 풀이 1

def solution(n):
    answer = 0
    for i in range(1, int(n ** 0.5) + 1):
        if n % i == 0:
            answer += 2

            if i * i == n:
                answer -= 1

    return answer

• 내 풀이2와 유사한데, 전체적으로 2씩 더하고 제곱근이면 1을 빼도록 만드셨다. 대체적으로 이러한 idea 를 사용한 풀이가 많았다.

다른 사람 풀이 2

def solution(n):
    return len(list(filter(lambda v: n % (v+1) == 0, range(n))))

정수 n이 매개변수로 주어질 때 n의 각 자리 숫자의 합을 return하도록 solution 함수를 완성해주세요

제한사항

0 ≤ n ≤ 1,000,000

idea

n을 문자열로 바꾼 후 하나씩 추출해서 더한다.

풀이 1

def solution(n):
    str_n = str(n)
    answer = 0
    for i in str_n:
        answer += int(i)
    return answer

풀이 2

def solution(n):
    return sum([int(i) for i in str(n)])

풀이 2가 풀이 1보다 속도가 느렸다. 굳이 리스트로 만들 필요는 없었다.

다른사람 풀이

def solution(n):
    return sum(list(map(int, str(n))))

집합(set)

1. 표현식

set(리스트 혹은 문자열)

## 2. 특징
> - 중복 허용 X
- 순서가 없음(Unordered)
- 교집합, 합집합, 차집합 연산 가능
- 확장함수: add, update, remove

## 3. 예제
> ```python
s1 = set([1, 1, 2, 3, 4, 5, 5, 5])
s1

# output
{1, 2, 3, 4, 5}

s2 = set("Hello")
s2

# output
{'e', 'l', 'o', 'H'}

4. 응용

s1 = set([1, 2, 2, 3, 3, 3, 4, 5])
s2 = set([4, 5, 5, 6])

### (1) 교집합
>```python
s1&s2

# output
{4, 5}

(2) 합집합

s1|s2

```python
# output
{1, 2, 3, 4, 5, 6}

(3) 차집합

s1 - s2

```python
# output
{1, 2, 3}

s2 - s1

# output
{6}

s1.difference(s2)

# output
{1, 2, 3}

s2.difference(s1)

# output
{6}

5. 관련 함수

(1) add

1개의 값만 추가

s1 = set([1, 2, 3])
s1.add(4)
s1

# output
{1, 2, 3, 4}

(2) update

값 여러개 추가

s1 = set([1, 2, 3])
s1.update([4, 5, 6])
s1

# output
{1, 2, 3, 4, 5, 6}

(3) remove

특정 값 제거

s1 = set([1, 2, 3])
s1.remove(2)
s1

# output
{1, 3}

출처 점프투파이썬

두 배열이 얼마나 유사한지 확인해보려고 합니다. 문자열 배열 s1과 s2가 주어질 때 같은 원소의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ s1, s2의 길이 ≤ 100
• 1 ≤ s1, s2의 원소의 길이 ≤ 10
• s1과 s2의 원소는 알파벳 소문자로만 이루어져 있습니다
• s1과 s2는 각각 중복된 원소를 갖지 않습니다.

풀이 1

• idea

  s1과 s2중에 길이가 더 짧은 애를 데리고와서, 그 리스트의 원소 각각이 다른 리스트에 있는지 반복문으로 확인

  def solution(s1, s2):
    if len(s1) < len(s2):
        a, b = s1, s2
    else:
        a, b = s2, s1
  
    answer = 0
    for i in a:
        if i in b:
            answer += 1
        else:
            continue
  
    return answer

풀이2

• idea

  s1, s2 두 리스트를 하나의 리스트로 합친 다음, 중복된 원소의 개수를 센다.

  def solution(s1, s2):
    s = s1 + s2
    set_s = set(s)
    return len(s)-len(set_s)

다른 사람의 풀이

def solution(s1, s2):
    return len(set(s1)&set(s2))

사분면은 한 평면을 x축과 y축을 기준으로 나눈 네 부분입니다. 사분면은 아래와 같이 1부터 4까지 번호를매깁니다.

• x 좌표와 y 좌표가 모두 양수이면 제1사분면에 속합니다.
• x 좌표가 음수, y 좌표가 양수이면 제2사분면에 속합니다.
• x 좌표와 y 좌표가 모두 음수이면 제3사분면에 속합니다.
• x 좌표가 양수, y 좌표가 음수이면 제4사분면에 속합니다.

x 좌표 (x, y)를 차례대로 담은 정수 배열 dot이 매개변수로 주어집니다. 좌표 dot이 사분면 중 어디에 속하는지 1, 2, 3, 4 중 하나를 return 하도록 solution 함수를 완성해주세요.

풀이

• IDEA 총 4개의 영역을 구분하는 if문을 작성한다.

  def solution(dot):
    x, y = dot[0], dot[1]
    print(x, y)
    if x>0 and y>0:
        return 1
    elif x<0 and y>0:
        return 2
    elif x<0 and y<0:
        return 3
    elif x>0 and y<0:
        return 4
    else:
        print("원점입니다")

다른 사람 풀이

def solution(dot):
    quad = [(3,2),(4,1)]
    return quad[dot[0] > 0][dot[1] > 0]

정수가 담긴 리스트 num_list가 주어질 때, num_list의 원소 중 짝수와 홀수의 개수를 담은 배열을 return 하도록 solution 함수를 완성해보세요.

제한사항

1 ≤ num_list의 길이 ≤ 100 0 ≤ num_list의 원소 ≤ 1,000

풀이 1

def solution(num_list):
    even_num = 0
    for num in num_list:
        if num % 2 == 0:
            even_num += 1
        else:
            continue
    return [even_num, len(num_list)-even_num]

• 답은 나오긴 하는데, 배열이 길수록 반복문 수행 시간이 오래걸릴테니까 다른 풀이를 생각해봄

풀이2 (최종제출)

• idea

  num_list 각 원소에 대해 2로 나눈 나머지를 구한 리스트를 만든 후, 0의 개수와 1의 개수를 각각 짝수의 개수, 홀수의 개수로 return한다.

def solution(num_list):
    answer = [num%2 for num in num_list]
    return [answer.count(0), answer.count(1)]

다른 사람 풀이

• idea

  [0, 0]이라는 길이 2짜리 리스트 answer를 만들어두고, 반복문으로 각 원소의 나머지가 0이라면 answer의 0번째 원소에 1을 더해주고, 나머지가 1이라면 answer의 1번째 원소에 1을 더해준다.

  def solution(num_list):
    answer = [0,0]
    for n in num_list:
        answer[n%2]+=1
    return answer

정수 n이 매개변수로 주어질 때, n 이하의 홀수가 오름차순으로 담긴 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ n ≤ 100

풀이

def solution(n):
    tmp = list(range(n+1))
    return [num for num in tmp if num%2 == 1]

다른사람 풀이

def solution(n):
    return [i for i in range(1, n+1, 2)]

// 연산자

tmp = 7 // 2
print(tmp)

# output
3

2. 나머지 구하기

% 연산자

tmp = 7 % 2
print(tmp)

# output
1

3. 몫과 나머지를 동시에 구하기

divmod(number1, number2) number1을 number2로 나눴을 때의 몫과 나머지를 튜플 형식으로 반환

tmp1, tmp2 = divmod(7, 2)
print('divmod(7, 2) =', divmod(7, 2))
print('tmp1 =', tmp1, '\ntmp2 =', tmp2)

# output
divmod(7, 2) = (3, 1)
tmp1 = 3 
tmp2 = 1

머쓱이는 추운 날에도 아이스 아메리카노만 마십니다. 아이스 아메리카노는 한잔에 5,500원입니다. 머쓱이가 가지고 있는 돈 money가 매개변수로 주어질 때, 머쓱이가 최대로 마실 수 있는 아메리카노의 잔 수와 남는 돈을 순서대로 담은 배열을 return 하도록 solution 함수를 완성해보세요.

제한사항

0 < money ≤ 1,000,000

풀이

def solution(money):
    price = 5500
    return [money // price, money % price]

다른 사람 풀이

def solution(money):
    return divmod(money, 5500)

문자열 배열 strlist가 매개변수로 주어집니다. strlist 각 원소의 길이를 담은 배열을 retrun하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ strlist 원소의 길이 ≤ 100 strlist는 알파벳 소문자, 대문자, 특수문자로 구성되어 있습니다.

풀이1

def solution(strlist):
    answer = []
    for i in strlist:
        answer.append(len(i))
    return answer

풀이2

def solution(strlist):
    return [len(i) for i in strlist]

정수 배열 numbers가 매개변수로 주어집니다. numbers의 각 원소에 두배한 원소를 가진 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

-10,000 ≤ numbers의 원소 ≤ 10,000 1 ≤ numbers의 길이 ≤ 1,000

풀이

def solution(numbers):
    answer = []
    for i in numbers:
        answer.append(i * 2)
    return answer

다른 사람 풀이 추가

def solution(numbers):
    return [num*2 for num in numbers]

[표현식 for 항목 in 반복가능객체 if 조건문]

1. 기본 예제

list_A 리스트의 각 항목에 3을 곱한 결과를 result 리스트에 담아보자.

list_A = [1, 2, 3, 4]

(1) 리스트 안에서 for문 사용하기

result = []
for num in list_A:
  result.append(num*3)
print(result)

# output
[3, 6, 9, 12]

(2) 리스트 내포 사용하기

result = [num*3 for num in list_A]
print(result)
# output
[3, 6, 9, 12]

2. 응용

(1) 리스트 내포 안에 if 조건 사용하기

[표현식 for 항목 in 반복가능객체 if 조건문]

result = [num*3 for num in list_A if num > 2]
print(result)
# output
[9, 12]

(2) for문 여러개

[표현식 for 항목1 in 반복가능객체1 if 조건문1  
        for 항목2 in 반복가능객체2 if 조건문2  
        ...  
        for 항목n in 반복가능객체n if 조건문n]

# 구구단 만들기 (1단, 2단)
result = [x*y for x in range(1, 3)
              for y in range(1,10)]
print(result)
# output
[1, 2, 3, 4, 5, 6, 7, 8, 9, 2, 4, 6, 8, 10, 12, 14, 16, 18]

출처 점프투파이썬

리스트를 문자열로 합쳐주는 함수

'구분자'.join(리스트)

예제

list_A = ['a', 'b', 'c', '1']
print(list_A)

-----output-----
['a', 'b', 'c', '1']

1. 리스트를 문자열로 변환하기

(1) join 이용

result1 = ''.join(list_A)
print(result1)

-----output-----
abc1

(2) 문자열을 하나씩 합치기

result2 = ''
for i in list_A:
    print("i: {0}".format(i))
    result2 += i
    print("result2: {0}".format(result2))

print("*\n final: {0}".format(result2))

-----output-----
i: a
result2: a
i: b
result2: ab
i: c
result2: abc
i: 1
result2: abc1
*
 final: abc1

2. 구분자 이용해보기

result3 = '_'.join(list_A)
print(result3)

-----output-----
a_b_c_1

result4 = '\n'.join(list_A)
print(result4)

-----output-----
a
b
c
1

출처 blockdmask님 블로그

정수 num1, num2가 매개변수로 주어질 때, num1을 num2로 나눈 몫을 return 하도록 solution 함수를 완성해주세요.

풀이

def solution(num1, num2):
    answer = num1 // num2
    return answer

문제 설명

정수 num1, num2가 매개변수 주어집니다. num1과 num2를 곱한 값을 return 하도록 solution 함수를 완성해주세요.

풀이

```python def solution(num1, num2): answer = num1*num2 return answer ``'

문제 설명

정수 num1과 num2가 주어질 때, num1에서 num2를 뺀 값을 return하도록 soltuion 함수를 완성해주세요.

풀이

def solution(num1, num2):
    answer = num1 - num2
    return answer

문제 설명

문자열 my_string과 정수 n이 매개변수로 주어질 때, my_string에 들어있는 각 문자를 n만큼 반복한 문자열을 return 하도록 solution 함수를 완성해보세요.

제한사항

2 ≤ my_string 길이 ≤ 5 2 ≤ n ≤ 10 "my_string"은 영어 대소문자로 이루어져 있습니다.

idea

빈 리스트에 반복문으로 0번째 원소부터 하나씩 선택해서 n 번 반복한 값을 계속 append한 후, string으로 변환해서 리턴하자

풀이

def solution(my_string, n):
    answer = []
    for i in my_string:
        answer.append(i * n)
    return ''.join(answer)

다른 사람 풀이

def solution(my_string, n):
    return ''.join(i*n for i in my_string)

문제 설명

머쓱이는 할머니께 생신 축하 편지를 쓰려고 합니다. 할머니가 보시기 편하도록 글자 한 자 한 자를 가로 2cm 크기로 적으려고 하며, 편지를 가로로만 적을 때, 축하 문구 message를 적기 위해 필요한 편지지의 최소 가로길이를 return 하도록 solution 함수를 완성해주세요.

제한사항

공백도 하나의 문자로 취급합니다. 1 ≤ message의 길이 ≤ 50 편지지의 여백은 생각하지 않습니다. message는 영문 알파벳 대소문자, ‘!’, ‘~’ 또는 공백으로만 이루어져 있습니다.

풀이

def solution(message):
    return len(message)*2

문제 설명

문자열 my_string과 문자 letter이 매개변수로 주어집니다. my_string에서 letter를 제거한 문자열을 return하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ my_string의 길이 ≤ 100 letter은 길이가 1인 영문자입니다. my_string과 letter은 알파벳 대소문자로 이루어져 있습니다. 대문자와 소문자를 구분합니다.

풀이

def solution(my_string, letter):
    return my_string.replace(letter, '')

문제 설명

선분 세 개로 삼각형을 만들기 위해서는 다음과 같은 조건을 만족해야 합니다.

가장 긴 변의 길이는 다른 두 변의 길이의 합보다 작아야 합니다. 삼각형의 세 변의 길이가 담긴 배열 sides이 매개변수로 주어집니다. 세 변으로 삼각형을 만들 수 있다면 1, 만들 수 없다면 2를 return하도록 solution 함수를 완성해주세요.

제한사항

sides의 원소는 자연수입니다. sides의 길이는 3입니다. 1 ≤ sides의 원소 ≤ 1,000

풀이

def solution(sides):
    sides.sort()

    if sides[2] < sides[0] + sides[1]:
        return 1
    else:
        return 2

다른 사람 풀이

def solution(sides):
    return 1 if max(sides) < (sum(sides) - max(sides)) else 2