An array A consisting of N integers is given. A triplet (P, Q, R) is triangular if 0 ≤ P < Q < R < N and:
- A[P] + A[Q] > A[R],
- A[Q] + A[R] > A[P],
- A[R] + A[P] > A[Q].
For example, consider array A such that:
A[0] = 10 A[1] = 2 A[2] = 5 A[3] = 1 A[4] = 8 A[5] = 20
Triplet (0, 2, 4) is triangular.
Write a function:
int solution(vector<int> &A);
that, given an array A consisting of N integers, returns 1 if there exists a triangular triplet for this array and returns 0 otherwise.
For example, given array A such that:
A[0] = 10 A[1] = 2 A[2] = 5 A[3] = 1 A[4] = 8 A[5] = 20
the function should return 1, as explained above. Given array A such that:
A[0] = 10 A[1] = 50 A[2] = 5 A[3] = 1
the function should return 0.
Write an efficient algorithm for the following assumptions:
- N is an integer within the range [0..100,000];
- each element of array A is an integer within the range [−2,147,483,648..2,147,483,647]
문제는 위와 같다.
조건은 저렇게 위에 세 개의 부등식으로 나와있지만, 정렬을해서 확인하면 작은 두 원소의 합이 가장 큰 원소보다 크면 성립하게 된다.
오버플로우 방지를 위해 하나의 항을 이항해서 계산해주면 된다.
1 2 3 4 5 6 7 8 9 10 11 12 | #include <algorithm> int solution(vector<int> &A) { // write your code in C++14 (g++ 6.2.0) if(A.size() < 3) return 0; sort(A.begin(), A.end()); for(int i = 0 ; i <= A.size()-3 ; i++){ if(A[i] > A[i+2]-A[i+1]) //오버플로우 방지 return 1; } return 0; } | cs |