[프로그래머스] 연속된 부분 수열의 합 / 투포인터 알고리즘

 

Level. 2

 

문제

비내림차순으로 정렬된 수열이 주어질 때, 다음 조건을 만족하는 부분 수열을 찾으려고 합니다.
기존 수열에서 임의의 두 인덱스의 원소와 그 사이의 원소를 모두 포함하는 부분 수열이어야 합니다.
부분 수열의 합은 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의 부분 수열로 만들 수 있는 값입니다.

 

 

풀이

#include <string>
#include <vector>
#include <iostream>
using namespace std;

vector<int> solution(vector<int> sequence, int k) {
    vector<int> answer = {0,0};
    int min_count = sequence.size()+1;
    int N = sequence.size();
    int count = 1;
    int sum = sequence[0];
    int s = 0, e = 0;
    
    while(e <= N){
        if(sum == k){
            if(count < min_count){
                answer[0] = s;
                answer[1] = e;
                min_count = count;
            }
        }
        
        if(sum <= k){
            e++;
            if(e<N){
                sum += sequence[e];
                count += 1;
            }
        }
        else if(sum > k){
            sum -= sequence[s];
            count -= 1;
            s += 1;
        }
    }
    
    return answer;
}

 

문제에서 부분수열을 찾으라고 했으므로, 투포인터알고리즘을 사용하여 부분수열을 구해주었다. 

 

투포인터 알고리즘 - Tow Pointers

• 리스트에 순차적으로 접근해야 할 때 두 점의 위치를 기록하면서 처리하는 알고리즘
• 특정한 합(N)을 가지는 부분수열을 찾을 때 주로 사용되는 알고리즘이다. 
  1. start와 end가 배열의 첫 번째 인덱스를 가리키도록 한다.
  2. 현재 부분수열의 합이 N 과 같다면 카운트한다.
  3. 현재 부분수열의 합이 N보다 작다면 end를 1 증가시킨다.
  4. 현재 부분 수열의 합이 N보다 크다면 start를 1 증가시킨다. 
  5. 모든 경우의 수를 확인할 때 까지 2~4 과정을 반복한다. 

 

투포인터 알고리즘 예시문제

- 프로그래머스 - 두 큐 합 같게 만들기

 

투포인터 알고리즘과 슬라이딩 윈도우 알고리즘의 차이 

• 투포인터 알고리즘 : 구간의 넓이가 조건에 따라 유동적으로 변하는 알고리즘
• 슬라이딩 윈도우 : 구간의 넓이가 고정되어 있어, 고정 사이즈의 윈도가 이동하면서 윈도우 내에 있는 데이터를 이용하는 알고리즘

슬라이딩 윈도우 알고리즘 예시 문제

- 프로그래머스 - 할인행사

 

 

 

 

 

---

다른 풀이

#include <vector>

using namespace std;

vector<int> solution(vector<int> sequence, int k)
{
    vector<int> answer;

    int srt=0, end=0, sum=0;

    for(int i=0; i<sequence.size(); i++)
    {
        end=i;
        sum+=sequence[end];

        while(sum>k)
            sum-=sequence[srt++];

        if(sum==k)
            if(answer.empty() || end-srt < answer[1]-answer[0])
                answer={srt, end};
    }

    return answer;
}

 

변수들이 확 줄어들었다!

 

if(answer.empty() || end-srt < answer[1]-answer[0])
		answer={srt, end};

 

위와 같은 방법을 사용하여 count와 min_count 변수를 따로 만들지 않아도 길이가 가장 짧은 수열을 찾을 수 있게 되었다. 

 

 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/178870?language=cpp#

프로그래머스