Given an integer array nums, return all the triplets [nums[i], nums[j], nums[k]] such that i != j, i != k, and j != k, and nums[i] + nums[j] + nums[k] == 0.

Notice that the solution set must not contain duplicate triplets.

 

Example 1:

Input: nums = [-1,0,1,2,-1,-4] Output: [[-1,-1,2],[-1,0,1]]

Example 2:

Input: nums = [] Output: []

Example 3:

Input: nums = [0] Output: []

 

Constraints:

  • 0 <= nums.length <= 3000
  • -105 <= nums[i] <= 105
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());
        int target, left, right, sum;
        for (int i = 0; i < nums.size(); i++)
        {
            if (i > 0 && nums[i - 1] == nums[i]) continue;
            
            target = -nums[i];
            left = i + 1;
            right = nums.size() - 1;
            while (left < right)
            {
                sum = nums[left] + nums[right];
                if (sum < target) left++;
                else if (sum > target) right--;
                else
                {
                    ans.push_back({ nums[i], nums[left], nums[right] });
                    while (left < right && nums[left] == ans[ans.size() - 1][1]) left++;
                    while (left < right && nums[right] == ans[ans.size() - 1][2]) right--;
                }
            }
        }
        return ans;
    }
};

'Challenge' 카테고리의 다른 글

[LeetCode] Letter Combinations of a Phone Number  (0) 2021.10.06
[LeetCode] 3Sum Closest  (0) 2021.10.06
[Programmers] 이분탐색 - 징검다리  (0) 2021.10.03
[Programmers] 이분탐색 - 입국심사  (0) 2021.10.03
[Codility] MinAvgTwoSlice  (0) 2021.10.03

출발지점부터 distance만큼 떨어진 곳에 도착지점이 있습니다. 그리고 그사이에는 바위들이 놓여있습니다. 바위 중 몇 개를 제거하려고 합니다.
예를 들어, 도착지점이 25만큼 떨어져 있고, 바위가 [2, 14, 11, 21, 17] 지점에 놓여있을 때 바위 2개를 제거하면 출발지점, 도착지점, 바위 간의 거리가 아래와 같습니다.

제거한 바위의 위치각 바위 사이의 거리거리의 최솟값

[21, 17] [2, 9, 3, 11] 2
[2, 21] [11, 3, 3, 8] 3
[2, 11] [14, 3, 4, 4] 3
[11, 21] [2, 12, 3, 8] 2
[2, 14] [11, 6, 4, 4] 4

위에서 구한 거리의 최솟값 중에 가장 큰 값은 4입니다.

출발지점부터 도착지점까지의 거리 distance, 바위들이 있는 위치를 담은 배열 rocks, 제거할 바위의 수 n이 매개변수로 주어질 때, 바위를 n개 제거한 뒤 각 지점 사이의 거리의 최솟값 중에 가장 큰 값을 return 하도록 solution 함수를 작성해주세요.

제한사항

  • 도착지점까지의 거리 distance는 1 이상 1,000,000,000 이하입니다.
  • 바위는 1개 이상 50,000개 이하가 있습니다.
  • n 은 1 이상 바위의 개수 이하입니다.
#include <vector>
#include <algorithm>

using namespace std;

int solution(int distance, vector<int> rocks, int n) {
    int answer = 0;
    int start = 1, end = distance;
    int mid, count, temp;
    sort(rocks.begin(), rocks.end());
    while (start <= end)
    {
        mid = (start + end) / 2;
        count = 0;
        temp = 0;
        for (int i = 0; i < rocks.size(); i++)
        {
            if (mid > (rocks[i] - temp)) ++count;
            else temp = rocks[i];
        }
        if (mid > (distance - temp)) ++count;
        if (count > n) end = (mid - 1);
        else
        {
            start = (mid+1);
            answer = mid;
        }
    }
    return answer;
}

'Challenge' 카테고리의 다른 글

[LeetCode] 3Sum Closest  (0) 2021.10.06
[LeetCode] 3Sum  (0) 2021.10.06
[Programmers] 이분탐색 - 입국심사  (0) 2021.10.03
[Codility] MinAvgTwoSlice  (0) 2021.10.03
[HackerRank] Diagonal Difference  (0) 2021.09.30

n명이 입국심사를 위해 줄을 서서 기다리고 있습니다. 각 입국심사대에 있는 심사관마다 심사하는데 걸리는 시간은 다릅니다.

처음에 모든 심사대는 비어있습니다. 한 심사대에서는 동시에 한 명만 심사를 할 수 있습니다. 가장 앞에 서 있는 사람은 비어 있는 심사대로 가서 심사를 받을 수 있습니다. 하지만 더 빨리 끝나는 심사대가 있으면 기다렸다가 그곳으로 가서 심사를 받을 수도 있습니다.

모든 사람이 심사를 받는데 걸리는 시간을 최소로 하고 싶습니다.

입국심사를 기다리는 사람 수 n, 각 심사관이 한 명을 심사하는데 걸리는 시간이 담긴 배열 times가 매개변수로 주어질 때, 모든 사람이 심사를 받는데 걸리는 시간의 최솟값을 return 하도록 solution 함수를 작성해주세요.

제한사항

  • 입국심사를 기다리는 사람은 1명 이상 1,000,000,000명 이하입니다.
  • 각 심사관이 한 명을 심사하는데 걸리는 시간은 1분 이상 1,000,000,000분 이하입니다.
  • 심사관은 1명 이상 100,000명 이하입니다.

입출력 예

ntimesreturn

6 [7, 10] 28

입출력 예 설명

가장 첫 두 사람은 바로 심사를 받으러 갑니다.

7분이 되었을 때, 첫 번째 심사대가 비고 3번째 사람이 심사를 받습니다.

10분이 되었을 때, 두 번째 심사대가 비고 4번째 사람이 심사를 받습니다.

14분이 되었을 때, 첫 번째 심사대가 비고 5번째 사람이 심사를 받습니다.

20분이 되었을 때, 두 번째 심사대가 비지만 6번째 사람이 그곳에서 심사를 받지 않고 1분을 더 기다린 후에 첫 번째 심사대에서 심사를 받으면 28분에 모든 사람의 심사가 끝납니다.

#include <vector>
#include <algorithm>

using namespace std;

long long solution(int n, vector<int> times) {
    long long answer = 0;
    sort(times.begin(), times.end());
    long long start = 1;
    long long end = (long long)times[times.size()-1] * n;
    long long mid, count;
    
    while (start <= end)
    {
        mid = (start + end) / 2;
        count = 0;
        for (int i = 0; i < times.size(); i++)
        {
            count += (mid / times[i]);
        }
        if (count < n)
        {
            start = (mid + 1);
        }
        else
        {
            end = (mid - 1);
            answer = mid;
        }
    }
    return answer;
}

'Challenge' 카테고리의 다른 글

[LeetCode] 3Sum  (0) 2021.10.06
[Programmers] 이분탐색 - 징검다리  (0) 2021.10.03
[Codility] MinAvgTwoSlice  (0) 2021.10.03
[HackerRank] Diagonal Difference  (0) 2021.09.30
[Programmers] 완전탐색 - 모의고사  (0) 2021.09.30

A non-empty array A consisting of N integers is given. A pair of integers (P, Q), such that 0 ≤ P < Q < N, is called a slice of array A (notice that the slice contains at least two elements). The average of a slice (P, Q) is the sum of A[P] + A[P + 1] + ... + A[Q] divided by the length of the slice. To be precise, the average equals (A[P] + A[P + 1] + ... + A[Q]) / (Q − P + 1).

For example, array A such that:

A[0] = 4 A[1] = 2 A[2] = 2 A[3] = 5 A[4] = 1 A[5] = 5 A[6] = 8

contains the following example slices:

  • slice (1, 2), whose average is (2 + 2) / 2 = 2;
  • slice (3, 4), whose average is (5 + 1) / 2 = 3;
  • slice (1, 4), whose average is (2 + 2 + 5 + 1) / 4 = 2.5.

The goal is to find the starting position of a slice whose average is minimal.

Write a function:

int solution(vector<int> &A);

that, given a non-empty array A consisting of N integers, returns the starting position of the slice with the minimal average. If there is more than one slice with a minimal average, you should return the smallest starting position of such a slice.

For example, given array A such that:

A[0] = 4 A[1] = 2 A[2] = 2 A[3] = 5 A[4] = 1 A[5] = 5 A[6] = 8

the function should return 1, as explained above.

Write an efficient algorithm for the following assumptions:

  • N is an integer within the range [2..100,000];
  • each element of array A is an integer within the range [−10,000..10,000].
int solution(vector<int> &A) {
    int ans = 0;
    float min = (A[0] + A[1]) / 2.f;
    float avg = 0;
    for (int i = 2; i < A.size(); i++)
    {
        avg = (A[i-1] + A[i]) / 2.f;
        if (avg < min)
        {
            min = avg;
            ans = (i-1);
        }
        avg = (A[i-2] + A[i-1] + A[i]) / 3.f;
        if (avg < min)
        {
            min = avg;
            ans = (i-2);
        }
    }
    return ans;
}

+ Recent posts