본문 바로가기
c++/프로그래머스

[프로그래머스] 수열과 구간 쿼리 4(c++)

by ilp 2023. 12. 25.
반응형

https://school.programmers.co.kr/learn/courses/30/lessons/181922

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr



#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, vector<vector<int>> queries) {
    for(int i=0;i<queries.size();i++){//첫버냊 for문
        for(int j=queries[i][0];j<=queries[i][1];j++){//두번쨰 for문
            if(j%queries[i][2]==0){
                arr[j]+=1;
            }
        }
    }

    return arr;
}

queries의 크기만큼 for문을 돌린다.

queries[i][0]부터 queries[i][1]까지 for문을 돌린다.

j가 queries[i][2]의 배수이면

arr[j]에 1을 더한다.

 

참고

처음 이문제를 풀떄

if(arr[j]%queries[i][2]==0){
      arr[j]+=1;
}

이렇게 작성해야 한다고 생각했다.

그러나 문제를 자세히 보니 j가 배수인지를 판단해야했다는 거다.

문제를 꼼꼼히 읽는게 중요하다.

배수인걸 판단할 떄 %를 사용했는데

a룰 b로 나눈 나머지가 0이면 a는 b의 배수이다.


반응형