1978번: 소수 찾기

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

www.acmicpc.net

✏️ 내가 작성한 코드 

n = int(input())
nlist = map(int, input().split())
cnt = 0

for num in nlist:
    if num == 1: # 1은 소수가 아니므로 continue
        continue
    else:
        for i in range(1,num+1):
            if i == num: # 밑에 else에서 break글 안거치고 끝까지 왔으므로 cnt 1증가
                cnt += 1
            else:
                if i != 1 and num%i == 0: # 1이 아닌 다른 수로 나눠지면 소수 아님
                    break
print(cnt)
 

1316번: 그룹 단어 체커

그룹 단어란 단어에 존재하는 모든 문자에 대해서, 각 문자가 연속해서 나타나는 경우만을 말한다. 예를 들면, ccazzzzbb는 c, a, z, b가 모두 연속해서 나타나고, kin도 k, i, n이 연속해서 나타나기 때

www.acmicpc.net

 

✏️ 내가 작성한 코드 

 

count = int(input())  # 입력 받을 단어 개수
words = []  # 입력 받은 단어들 넣을 배열
group_word_cnt = 0  # 그룹 단어 개수

for _ in range(count): # 단어들 입력 받음
    word = input()
    words.append(word)

for word in words:
    chr_list = [word[0]] # 단어의 각 문자 들어갈 배열, 첫문자 일단 넣어줌
    if len(word) == 1:
        group_word_cnt += 1
    else:
        for i in range(1,len(word)):
            if word[i] == word[i-1]: # 앞 문자랑 같은 경우
                if i == len(word)-1: # 앞 문자랑 같고, 현재 인덱스가 마지막 인덱스라면
                    group_word_cnt += 1 # 그룹단어 해당
                continue # 앞 문자랑 같으면 continue
            else: # 앞 문자랑 같지 않은 경우
                if word[i] in chr_list : # 앞 문자랑 다르고 배열에 있으면 --> 그룹단어아님
                    break
                chr_list.append(word[i]) # 앞 문자랑 다르고 배열에 없으면 배열에 넣어줌
            if i == len(word)-1: # 마지막 인덱스까지 반복문 거쳐서 온 경우
                group_word_cnt += 1 # 그룹단어 해당

print(group_word_cnt)
 

2606번: 바이러스

첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어

www.acmicpc.net

 

✏️ 내가 작성한 코드 (BFS 사용)

# 조건 : 1번 컴퓨터가 바이러스에 걸림

def bfs(start):
    visit = 0 # 바이러스 걸린 컴퓨터를 셀 변수
    need_visit= [] # 큐 대신 리스트 사용
    need_visit.append(start) # 시작 컴퓨터 처음에 넣어줌

    while need_visit: # 리스트에 있는 값 모두 없어질때까지 while 돌아감
        node = need_visit.pop(0) # 큐 특성 이용 -> 맨 앞값 가져옴 *pop() : 맨뒷값 가져옴
        if not visited[node]:
            visited[node]=True
            if node != 1: # 1번 제외하고 횟수 세야함
                visit +=1
            for e in abj[node]: # 해당 컴퓨터에 연결된 컴퓨터들 돌아가면서 가져오고 리스트에 넣어줌
                if not visited[e]:
                    need_visit.append(e)

    return visit


start = 1 # 시작 컴퓨터 번호
computer = int(input()) # 컴퓨터의 수
connec = int(input()) # 연결되어 있는 컴퓨터 쌍의 수
abj = [[] for _ in range(computer+1)] # 연결된 컴퓨터 쌍 담을 리스트
visited = [False]*(computer+1)

for _ in range(connec):
    a,b = map(int,input().split()) # 한 쌍씩 연결되어 있는 컴퓨터 번호 주어짐
    abj[a].append(b) # 양방향이므로 각각 append해줌줌    abj[b].append(a)
    abj[b].append(a)

print(bfs(start))

 

✏️ 참고

'from collections import deque' 사용하여 큐를 이용하는 대신

리스트를 이용. pop(0) 써서 맨 앞값 가져옴

+ Recent posts