연속된 부분 수열의 합

문제 설명

비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.

 

기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.

부분 수열의 합은 k입니다.

합이 k인 부분 수열이 여러 개인 경우 길이가 짧은 수열을 찾습니다.

길이가 짧은 수열이 여러 개인 경우 앞쪽(시작 인덱스가 작은)에 나오는 수열을 찾습니다.

수열을 나타내는 정수 배열 sequence와 부분 수열의 합을 나타내는 정수 k가 매개변수로 주어질 때, 위 조건을 만족하는 부분 수열의 시작 인덱스와 마지막 인덱스를 배열에 담아 return 하는 solution 함수를 완성해주세요. 이때 수열의 인덱스는 0부터 시작합니다.

 

제한사항

• 5 ≤ sequence의 길이 ≤ 1,000,000
• 1 ≤ sequence의 원소 ≤ 1,000
• sequence는 비내림차순으로 정렬되어 있습니다.
• 5 ≤ k ≤ 1,000,000,000
• k는 항상 sequence의 부분 수열로 만들 수 있는 값입니다.

입출력 예

sequence	k	result
[1, 2, 3, 4, 5]	7	[2, 3]
[1, 1, 1, 2, 3, 4, 5]	5	[6, 6]
[2, 2, 2, 2, 2]	6	[0, 2]

 

---

문제 해설

 

쉽게 이해해보자. 길을 가고있는 지렁이가 있고, 그 길은 전부 음식으로 체워져있다.

 

지렁이는 언제나 앞으로 이동한다.

 

지렁이는 앞으로 이동할 때 무조건 앞에있는 음식을 먹는다.

 

지렁이는 먹은 음식 그대로 몸의 크기가 되어버리는 마법의 지렁이라, 먹을것을 먹으면 꼬리가 움직이지 않아도 머리가 앞으로 갈 수 있다.

 

그런데, 전체 몸의 크기는 제한되어있어서, 일정크기를 넘어가게되면 먹은걸 뱉어버린다.

 

몸의크기는 제한되어있기 때문에 지렁이가 앞으로 가게되면, 꼬리가 앞으로 딸려오게된다.

 

이렇게 지렁이가 길의 끝까지 이동할 때, 그 지렁이의 크기가 가장 작을 때 를 구하는 문제이다.

---

첫 번째 시도

#include <string>
#include <vector>
#include <queue>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> sequence, int k) {
    vector<int> answer(2);
    answer[0] = 0;
    answer[1] = sequence.size() - 1;

    queue<int> inputQ;
    inputQ.push(0);

    int nowpos = 1;
    int nowsize = sequence[0];

    while (inputQ.size() != 0)
    {
        bool chack = true;
        if (nowsize < k && nowpos < sequence.size()) 
        {
            nowsize += sequence[nowpos];
            inputQ.push(nowpos);
            nowpos++;
            chack = false;
        }
        else if (nowsize == k)
        {
            if (answer[1] - answer[0] > (nowpos - 1) - inputQ.front())
            {
                answer[1] = nowpos - 1;
                answer[0] = inputQ.front();
            }
        }

        if(chack)
        {
            nowsize -= sequence[inputQ.front()];
            inputQ.pop();
        }
    }
    return answer;
}

성공

 

큐를 사용해서 쭉 값을 밀어넣다가 적절한 데이터가 발견되었을 때, 그걸 result로 기입할지 말지 판단하고 넣는다.

 

앞으로 쭉 가다가 넣지못하는 값을 만나게 되면 지렁이는 굶어죽는다.