본문 바로가기
코드트리 알고리즘/Python

[코드트리] 함수를 이용한 369 게임 - Python

by 리버🐦‍🔥 2023. 8. 31.

<문제>

https://www.codetree.ai/missions/5/problems/369-games-using-functions/introduction

 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

국가대표가 만든 코딩 공부의 가이드북 코딩 왕초보부터 꿈의 직장 코테 합격까지, 국가대표가 엄선한 커리큘럼으로 준비해보세요.

www.codetree.ai

<풀이>

1. string.find()를 사용했을 때, 값이 포함되어있지 않으면 -1을 리턴하는 성질을 활용해 '3', '6', '9'일 때를 처리

2. i % 3을 통해 배수인지 판단 후 처리

 

<코드>

def func1(a, b):
    cnt = 0
    for i in range(a, b + 1):
        if str(i).find("3") != -1 or str(i).find("6") != -1 or str(i).find("9") != -1 or i % 3 == 0:
            cnt += 1
    return cnt

a, b = map(int, input().split())

print(func1(a, b))