Cộng đồng Hỏi & Đáp của Inflearn

밑에 코드와 같이 강의를 듣기 전에 먼저 코드를 작성해보았습니다. 선생님 강의 내용에서는 함수를 사용했지만 저는 함수 대신 다른 방법으로 짰는데 모두 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;

}