โปรแกรม C++ ตรวจสอบการจัดเรียงอาร์เรย์ให้สูตรคืนค่า m

คำอธิบายปัญหา

สมมติว่าเรามีอาร์เรย์ A n องค์ประกอบ และตัวเลข m ต้องการตรวจสอบว่าจัดเรียงอาร์เรย์ใหม่ได้ไหมที่: $$\mathrm{\sum_{i=1}^{n} \sum_{j=1}^{n}\frac{A[j]}{j} = m}$$ การหาร A[j]/j ไม่มีการปัดเศษ

ตัวอย่าง: A = [2, 5, 1]; m = 8 ผลลัพธ์คือ True เพราะจัดเป็น [1, 2, 5] ได้ (1/1 + 2/2 + 5/3) + (2/2 + 5/3) + (5/3) = 8

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

sum := 0
n := ขนาดของ A
สำหรับ i := 0 ถึง n-1:
    sum := sum + A[i]
ถ้า sum เท่ากับ m: คืนค่า true
มิฉะนั้น: คืนค่า false

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

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

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

int main() {
    vector<int> A = { 2, 5, 1 };
    int m = 8;
    cout << solve(A, m) << endl;
}

อินพุต

{ 2, 5, 1 }, 8

เอาต์พุต

1