방형구 탐색 (Easy) (백준 30503)

안녕하세요. 오늘은 꽃의 개수를 셀 거예요.

문제

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

아이디어

L과 R이 입력되면 그 범위를 다 훑어보면서 0으로 바꾸거나 꽃의 개수를 세거나를 반복해주면 됩니다.

소스코드

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

int main(void)
{
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    int N, i, j, arr[1010] = { 0 }, Q, l, r, k, cnt;

    cin >> N;
    for (i = 1; i <= N; i++) cin >> arr[i];
    cin >> Q;
    for (i = 0; i < Q; i++)
    {
        int type;
        cin >> type >> l >> r;
        if (type == 1)
        {
            cnt = 0;
            cin >> k;
            for (j = l; j <= r; j++)
                if (arr[j] == k)
                    cnt++;
            cout << cnt << "\n";
        }
        else
        {
            for (j = l; j <= r; j++)
                arr[j] = -1;
        }
    }
}

감사합니다.