인프런 커뮤니티 질문&답변

밑에 코드와 같이 강의를 듣기 전에 먼저 코드를 작성해보았습니다. 선생님 강의 내용에서는 함수를 사용했지만 저는 함수 대신 다른 방법으로 짰는데 모두 wrong answer이 나오네요ㅠㅠ

계속해서 어디 부분이 틀렸는지 보지만 알고리즘은 맞는 듯해보이는데 혹시 어디서 오류가 발생했나요?

#include <iostream>

#include <vector>

#include <algorithm>

#include <string>

using namespace std;

int main() {

int n, m, tmp, cnt=0, mid = 0, start = 1, end;

scanf("%d %d",&n,&m);

vector <int> array(n);

for (int i = 0; i < n; i++){

scanf("%d",&array[i]);

end = end + array[i];

}

while (start <= end){

int pl = 0;

mid = (start + end) / 2;

for (int i = 0; i < n; i++){

pl = pl + array[i];

if (pl > mid) {

cnt++;

pl = array[i];

}

}

if (cnt <= m) {

end = mid - 1;

tmp = mid;

}

else start = mid + 1;

}

printf("%d",tmp);

return 0;

}