반응형 알고리즘/수학26 [백준][Python] 4134번 다음 소수 4134번: 다음 소수 첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다. www.acmicpc.net 코드 import math def prime(a): for i in range(2,int(math.sqrt(a))+1): if a % i == 0: return False # 소수 아님 return True num = int(input()) for i in range(num): a = int(input()) if a < 2: print(2) else: while 1: if prime(a): print(a) break a+=1 설명 math.sqrt()를 이용하여 소수인지 확인하는 범위를 해당 숫자의 제곱근으로 줄이면 시간이 단축된다. 주어진 숫.. 2022. 9. 13. [백준][Python] 21312번 홀짝 칵테일 21312번: 홀짝 칵테일 정진이는 특별한 음료를 가지고 있다. 음료들은 정수로 표현되는 고유 번호를 가지고 있다. 정진이는 이 음료들을 섞어 만든 칵테일을 만든다. 이 칵테일은 홀짝 칵테일이라 부르는데, 홀짝 칵 www.acmicpc.net 코드 n_list = list(map(int,input().split())) odd = [] for i in range(3): if (n_list[i] %2) != 0: odd.append(n_list[i]) ans = 1 if not odd: for i in range(3): ans *= n_list[i] else: for i in range(len(odd)): ans *= odd[i] print(ans) 설명 입력 받은 수가 홀수이면 odd 리스트에 넣어놓는다... 2022. 5. 19. [백준][Python] 1747번 소수&팰린드롬 1747번: 소수&팰린드롬 어떤 수와 그 수의 숫자 순서를 뒤집은 수가 일치하는 수를 팰린드롬이라 부른다. 예를 들어 79,197과 324,423 등이 팰린드롬 수이다. 어떤 수 N (1 ≤ N ≤ 1,000,000)이 주어졌을 때, N보다 크거나 같고, www.acmicpc.net 코드 import math N = int(input()) def prime(N): for i in range(2,int(math.sqrt(N))+1): if N%i == 0: return False return True ans = 0 for i in range(N,1000001): if i>1 and str(i) == str(i)[::-1]: if prime(i): print(i) ans = 1 break if ans == .. 2022. 3. 22. [백준][Python] 1934번 최소공배수 1934번: 최소공배수 두 자연수 A와 B에 대해서, A의 배수이면서 B의 배수인 자연수를 A와 B의 공배수라고 한다. 이런 공배수 중에서 가장 작은 수를 최소공배수라고 한다. 예를 들어, 6과 15의 공배수는 30, 60, 90등이 있 www.acmicpc.net 코드 import sys import math t = int(input()) for _ in range(t): a,b = map(int,sys.stdin.readline().split()) print(math.lcm(a,b)) 2022. 3. 21. 이전 1 2 3 4 5 6 7 다음 반응형