터널 속의 기차

1. 문제 풀이 2. 읽은 그대로 3. 더 어렵게 나옴...

#include <stdio.h> #define MAX 15 int N, H, l[MAX], d[MAX]; int main() { freopen("input.txt", "r", stdin); int T; scanf("%d", &T); for (int tc = 1; tc <= T; tc++) { scanf("%d %d", &N, &H); for (int i = 0; i < N; i++) scanf("%d", l + i); for (int i = 0; i < N; i++) scanf("%d", d + i); int ans = (d[0] == 0); // 처음에 불이 안켜있으면, 키고 시작해야함 d[0] = 1; if (d[N - 1] == 0) ans++; // 제일 마지막도 마찬가지 d[N - 1] = 1; int nxt = H; for (int i = 1; i < N; i++) { if (d[i]) nxt = H; else { nxt -= l[i]; if (nxt <= 0) { nxt = H; // 다음 기차를 체크하기 위함 ans++; } } } printf("#%d %d\n", tc, ans); } return 0; }