Olinfo/pens.cpp at main · Nataso08/Olinfo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#include <iostream>
#include <fstream>

using namespace std;

int main() {
//  uncomment the following lines if you want to read/write from files
	ifstream cin("input.txt");
 	ofstream cout("output.txt");

	int N;
    cin >> N;

    int pens[N];
    for(int i=0; i<N; i++)
        cin >> pens[i];

    // insert your code here
    int ans = 0;
    bool f;

    for (int i=0; i<N; i++) {
		if (pens[i] == 0) {
			if (i != 0 && i != N-1) {
				f = false;
				if (pens[i-1] > 1) {
					pens[i-1]--;
					f = true;
				}

				if (pens[i+1] > 1 && !f) {
					pens[i+1]--;
					f = true;
				}

				if (!f) {
					ans ++;
				}
			} else if (N != 1) {
				if (i == 0) {
					if (pens[i+1] > 1) {
						pens[i+1]--;
					} else {
						ans ++;
					}
				} else if (i == N-1) {
					if (pens[i-1] > 1) {
						pens[i-1]--;
					} else {
						ans ++;
					}
				}
			} else {
				ans = 1;
			}

		}

	}

    cout << ans << endl; // change 42 with actual answer
    return 0;
}