백준 문제 링크
지뢰
❓접근법
- 지뢰가 1개일 때, 여러 개일 때로 나눠야한다.
1개일 때는 1번을 출력하면 된다.- 여러 개일 때는 다음 조건을 충족시켜야 한다.
- 만약 인덱스가 0일 때
그 다음 지뢰의 강도보다 크거나 같다면 1을 출력한다.- 만약 인덱스가 N-1일 때
그 전 지뢰의 강도보다 크거나 같다면 N을 출력한다.- 그 외에는
그 전 지뢰의 강도와 그 다음 지뢰의 강도보다 모두 크거나 같다면 i+1을 출력한다.
- 여기서 주의해야 할 점은 문제에서 '각각의 지뢰는 충격 강도 Pi가 있어서, Pi를 초과하는 힘을 가하면 Pi만큼의 힘을 발휘하며 터지게 된다.'라는 점에 주의해야 한다.
예를 들어 5개의 지뢰가 있고 각각 [1, 1, 1, 1, 1] 이라고 생각해보자.
나는 처음에는 2번, 4번을 터트리면 다 터지는거 아닌가 라고 생각했는데
그게 아니라 충격 강도가 같다면 연쇄 반응을 중단하기 때문에
1번, 2번, 3번, 4번, 5번 전부 다 터트려야되는 것이다.
👌🏻코드
N = int(input())
data = []
for _ in range(N):
data.append(int(input()))
if N == 1:
print(data[0])
else:
for i in range(N):
if i == 0:
if data[i] >= data[i+1]:
print(i+1)
elif i == N-1:
if data[i] >= data[i-1]:
print(i+1)
else:
if data[i] >= data[i-1] and data[i] >= data[i+1]:
print(i+1)
.