백준16162 - 가희와 3단 고음

https://www.acmicpc.net/problem/16162

16162번: 가희와 3단 고음

그리디를 이용하여 풀었다. 초항을 먼저 찾아 벡터에 넣은 후 공차 만큼 더한 숫자를 차례대로 찾아가서 해당하는 수를 차곡차곡 벡터에 넣어서 길이를 출력하면 된다.

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int N, A, D;
    int x;
    int next;
    vector<int> v;
    
    cin >> N >> A >> D;
    for(int i = 0; i < N; i++) {
        cin >> x;
        if(v.size() == 0 && x == A) {
            v.push_back(x);
            next = A + D;
        }
        else if(x == next) {
            v.push_back(x);
            next = next + D;
        }
    }
    
    cout << v.size();
    return 0;
}