주어진 수식이 m을 반환하는지 여부와 같은 방식으로 배열을 재정렬할 수 있는지 확인하는 C++ 프로그램

n개의 요소와 다른 숫자 m이 있는 배열 A가 있다고 가정합니다. 다음과 같은 방식으로 배열을 재배열할 수 있는지 확인해야 합니다.

$$\mathrm{\sum_{i=1}^{n} \sum_{j=1}^{n}\frac{A[j]}{j} =m}$$

A[j]/j 연산에는 반올림이 없습니다.

따라서 입력이 A =[2, 5, 1]과 같으면; m =8이면 [1, 2, 5], (1/1 + 2/2 + 5/3) + (2/2 + 5/3) + (5) 배열의 경우 출력이 True가 됩니다. /3) =8

단계

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

sum := 0
n := size of A
for initialize i := 0, when i < n, update (increase i by 1), do:
   sum := sum + A[i]
if sum is same as m, then:
   return true
Otherwise
   return false

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#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