โค้ด C++ นับการดำเนินการทำให้สองอาร์เรย์เหมือนกัน

สมมติว่าเรามีอาร์เรย์ A และ B ขนาด n พิจารณาการดำเนินการ: เลือกดัชนี i และ j ลดค่า i ลง 1 เพิ่มค่า j ขึ้น 1 ทุกองค์ประกอบต้องไม่ติดลบหลังดำเนินการ ต้องการทำให้ A และ B เหมือนกัน หาลำดับการดำเนินการ ถ้าไม่ได้ส่ง -1

เช่น ถ้าอินพุตคือ A = [1, 2, 3, 4]; B = [3, 1, 2, 4] เอาต์พุตจะเป็น [(1, 0), (2, 0)] เพราะ i=1, j=0 ได้ [2,1,3,4] แล้ว i=2, j=0 ได้ [3,1,2,4]

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

a := 0, b := 0, c := 0
n := ขนาดของ A
กำหนดอาร์เรย์ C ขนาด n เติม 0
for i := 0 ถึง n-1: a := a + A[i]
for i := 0 ถึง n-1: b := b + A[i]
if a != b: return -1
otherwise:
    for i := 0 ถึง n-1:
        c := c + |A[i] - B[i]|
        C[i] := A[i] - B[i]
    c := c / 2
    i := 0, j := 0
    while c > 0:
        while C[i] <= 0: i++
        while C[j] >= 0: j++
        print i และ j
        ลด C[i] เพิ่ม C[j] ขึ้น 1
        c--

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

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

void solve(vector<int> A, vector<int> B) {
    int a = 0, b = 0, c = 0;
    int n = A.size();
    vector<int> C(n, 0);
    for (int i = 0; i < n; i++) a += A[i];
    for (int i = 0; i < n; i++) b += A[i];
    if (a != b) {
        cout << -1;
        return;
    } else {
        for (int i = 0; i < n; i++) {
            c += abs(A[i] - B[i]);
            C[i] = A[i] - B[i];
        }
        c = c / 2;
        int i = 0, j = 0;
        while (c--) {
            while (C[i] <= 0) i++;
            while (C[j] >= 0) j++;
            cout << "(" << i << ", " << j << "), ";
            C[i]--, C[j]++;
        }
    }
}

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

อินพุต

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

เอาต์พุต

(1, 0), (2, 0),