けんちょんの競プロ精進記録

競プロの精進記録や小ネタを書いていきます

AtCoder ABC 170 A - Five Variables (灰色, 100 点)

これ結構難しいと思うけど、Difficulty 8 なのすごい!

問題概要

整数  x_{1}, x_{2}, x_{3}, x_{4}, x_{5} が与えられる。このうち 1 つだけ 0 がある。

 x_{i} = 0 であるような  i を答えよ。

解法

if 文で 1 個ずつ見ていってもいいが、for 文でまとめて扱った方が楽だと思う。

#include <bits/stdc++.h>
using namespace std;

int main() {
    int res = -1;
    for (int i = 0; i < 5; ++i) {
        int x;
        cin >> x;
        if (x == 0) res = i + 1;
    }
    cout << res << endl;
}