โค้ด C++ ตรวจสอบจัดเรียง B ใหม่ให้ผลรวมไม่เกิน x

สมมติว่าเรามีอาร์เรย์ A และ B ขนาด n และตัวเลข x ต้องตรวจสอบว่าจัดเรียง B ใหม่ได้ไหม ที่ทำให้ A[i] + B[i] <= x สำหรับทุก i 0 ถึง n-1

เช่น ถ้าอินพุตคือ A = [1, 2, 3]; B = [1, 1, 2]; x = 4 เอาต์พุตจะเป็น True เพราะจัด B เป็น [1, 2, 1] ได้ 1+1<=4, 2+2<=4, 3+1<=4

ขั้นตอนการแก้ปัญหา

n := ขนาดของ A
ans := 1
sum := 0
for i := 0 ถึง n-1:
    sum := A[i] + B[n - i - 1]
    if sum > x: ans := 0
if ans: return true
otherwise: return false

ตัวอย่างโค้ด

#include <bits/stdc++.h>
using namespace std;

bool solve(vector<int> A, vector<int> B, int x) {
    int n = A.size();
    int ans = 1;
    int sum = 0;
    for (int i = 0; i < n; ++i) {
        sum = A[i] + B[n - i - 1];
        if (sum > x) ans = 0;
    }
    if (ans) return true;
    else return false;
}

int main() {
    vector<int> A = { 1, 2, 3 };
    vector<int> B = { 1, 1, 2 };
    int x = 4;
    cout << solve(A, B, x) << endl;
}

อินพุต

{ 1, 2, 3 }, { 1, 1, 2 }, 4

เอาต์พุต

1