인프런 커뮤니티 질문&답변
작성한 질문수
Section3 - 5번 문제 수들의 합 질문입니다.
작성
·
346
0
# 수들의 합
import sys
sys.stdin=open("input.txt","r")
n,m=map(int,input().split())
a=list(map(int,input().split()))
result=0
for i in range(n):
sum=a[i]
print("a[i]",a[i])
for j in range(i+1,n):
if sum==m:
result+=1
break
elif sum<m:
sum = sum + a[j]
else :
break
print(result)
이 문제 관련해서 전 이런식으로 풀었는데, in 2와 in 4가 제가 출력한 값과 1씩 차이가 났습니다. ㅠㅠ 이유를 도저히 모르겠어서 질문을 드립니다. 나머지 in 1,3,5는 정상적으로 동작합니다. 어느 로직에서 잘못이 있었을까요?!
* 2중 for문으로 인한 시간복잡도는 수정할 예정입니다!
- 수업 잘 듣고 있습니다! 감사합니다.
답변 1
0
김태원
지식공유자
안녕하세요^^
위 코드는 수열의 마지막 부분에서 합이 m이 되는 경우를 카운팅하지 못합니다.
아래 코드처럼 출력해 보세요
# 수들의 합
import sys
sys.stdin=open("in1.txt","r")
n,m=map(int,input().split())
a=list(map(int,input().split()))
result=0
for i in range(n):
sum=a[i]
for j in range(i+1,n):
if sum==m:
result+=1
print(i, j)
break
elif sum<m:
sum = sum + a[j]
else :
break
print(result)저라면 아래와 같이 하겠습니다. 비교해보세요.
import sys
#sys.stdin=open("in1.txt","r")
n,m=map(int,input().split())
a=list(map(int,input().split()))
result=0
for i in range(n):
sum=0
for j in range(i,n):
sum+=a[j]
if sum==m:
result+=1
#print(i, j)
break
if(sum>m):
break
print(result)