CSES - Sum of Two Values | Tổng hai giá trị

Tác giả:

Dạng bài

Điểm: 900 Thời gian: 1.0s Bộ nhớ: 512M Input: bàn phím Output: màn hình

Bạn được cho một mảng gồm \(n\) số nguyên và nhiệm vụ của bạn là tìm hai giá trị (tại các vị trí phân biệt) có tổng là \(x\).

Input

Dòng đầu vào đầu tiên có hai số nguyên \(n\) và \(x\): kích thước mảng và tổng mong muốn.
Dòng thứ hai có \(n\) số nguyên \(a_1,a_2,\ldots,a_n\): các giá trị của mảng.

Output

In hai số nguyên: vị trí của các giá trị. Nếu có một số lời giải, bạn có thể in bất kỳ lời giải nào trong số đó. Nếu không có lời giải nào, in IMPOSSIBLE.

Constraints

\(1 \leq n \leq 2 \cdot 10 ^ 5\)
\(1 \leq x, a_i \leq 10 ^ 9\)

Example

Sample input

4 8
2 7 5 1

Sample output

2 4

Bình luận

include<bits/stdc++.h>

define ll long long

using namespace std;

const int maxN = 200005;
int n, k;
ll a[maxN], b[maxN];

int indexx = 0;
int findPosition(ll arr[], ll x) {
for (int i = 0; i < n; i++) {
if (arr[i] == x && i != indexx - 1) {
return i + 1;
}
}
return 0;
}

void find() {
sort(a, a + n);
for (int i = 0; i < n; i++) {
ll al = a[i];
ll ar = k - al;

    int left = i + 1;
    int right = n - 1;
    while (left <= right) {
        int mid = (left + right) / 2;
        if (a[mid] == ar) {
            int idxl = findPosition(b, al);
            indexx = idxl;
            int idxr = findPosition(b, ar);
            cout << idxl << " " << idxr << "\n";
            return;
        }
        if (a[mid] > ar) {
            right = mid - 1;
        } else {
            left = mid + 1;
        }
    }
}
cout << "IMPOSSIBLE";

}

int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);

cin >> n >> k;
for (int i = 0; i < n; i++) {
    cin >> a[i];
    b[i] = a[i];
}

find();
return 0;

}

4 bình luận nữa