[백준] 합이 0

3Sum 문제를 풀고 이 문제를 풀려니깐 되게 쉽다고 느껴져서 거침없이 풀었다가 시간초과 + 틀렸습니다를 연속으로 마주하게 되어서 슬펐던 문제다. 3Sum 문제와 유사하게 세 숫자의 합이 0이 되는 조합을 발견하면 Answer 을 하나씩 올려주는 문제인데 조금 다른게 있다.

중복이 허용되는 문제이고 만약에 숫자가 같더라도 index 만 다르다면 아예 다른 류의 숫자로 구분이 되었다.

예시) -10 2 2 2 8 8 8 8

얼핏보면은, {-10,2,8}, {-10,2,8}, {-10,2,8} 아닌가? 라고 생각할수 있지만 사실 저 예시에서 나올 수 있는 조합은 12가지이다. {-10,2,8} 이 하나의 조합이고 다른 위치에 8들이 전부 저 위치에 들어갈수있다. 생각보다 수학적 사고방식이 들어가야 했던 문제고 좀 까다롭게 풀었다. 그냥 무지성으로 while 룹으로 계산하는것조차 안되었다.

예시) -10 5 5 5 5 5 5 5

이런 조합에서는 nC2 라는 콤비네이션 계산을 통해서 조합을 가지고 와야했다. 왜냐면 남은 5의 숫자 중에서 단 두개만을 골라서 콤비네이션을 만드는것이기 때문이다.

answer 또한 int 로 두면은 정답이 아니라고 나오고 꼭 long 으로 바꿔줬어야했다.

개인적으로 문제를 쓸데없이 많으 꼬아낸거같고 별로 맘에는 들지 않는 문제이다.

#include <iostream> 
#include <bits/stdc++.h> 
#define endl "\n"
#define MAX 100010
using namespace std;

int N; 
vector<int> nums; 

void Input(){
    cin >> N;
    for(int i = 0; i < N; i++){
        int n; 
        cin >> n; 
        nums.push_back(n); 
    }
}

int comb(int n){
    return n * (n-1)/2; 
}

void Solution(){
    long answer = 0; 
    sort(nums.begin(),nums.end()); 
    for(int i = 0; i < nums.size()-2; i++){
        if(nums[i] > 0) break; 
        int start = i + 1, end = nums.size()-1; 
        while(start < end){
            int s = 1;
            int e = 1; 
            int total = nums[i] + nums[start] + nums[end]; 
            if(total == 0){
                if(nums[start] == nums[end]){
                    answer += comb(end - start + 1);
                    break; 
                }
                while(start + 1 < end && nums[start] == nums[start+1]){
                    s++;
                    start++; 
                }
                while(end -1 > start && nums[end] == nums[end-1]){
                    e++;
                    end--; 
                }

                answer += s * e; 
            }
            if(total > 0){
                end--; 
            }
            else {
                start++; 
            }
        }
    }
    cout << answer; 
}


void Solve(){
    Input();
    Solution(); 
}

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    //freopen("input.txt", "r", stdin);
    Solve();
    return 0;

}

배운점:
1. 설명을 잘 읽어야한다.
2. 투포인터라고 꼭 같은 투포인터가 아니다