C++에서 N개의 고유 정수 합계를 0으로 찾기

정수 n이 있다고 가정합니다. n개의 고유한 정수를 포함하는 배열을 반환해야 합이 0이 됩니다. 따라서 입력이 n =5인 경우 가능한 출력은 [-7, -1, 1, 3, 4]

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

• 배열 A를 최종 답으로 취하고 x :=0
• 0 ~ n – 2 범위의 i에 대해
  • A[i] =(i + 1)
  • x :=x + i + 1
• A[n – 1] =x
• 리턴 A

예

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

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<int> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
class Solution {
   public:
      vector<int> sumZero(int n) {
         vector <int> ans(n);
         int x = 0;
         for(int i = 0; i < n - 1; i++){
            ans[i] = (i + 1);
            x += (i + 1);
         }
         ans[n - 1] = -x;
         return ans;
      }
};
main(){
   Solution ob;
   print_vector(ob.sumZero(10)) ;
}

입력

10

출력

[1, 2, 3, 4, 5, 6, 7, 8, 9, -45, ]