Computer >> 컴퓨터 >  >> 프로그래밍 >> C++

C++로 두 배열의 요소별 합을 구해 자릿수를 새 배열에 저장하는 방법

이 튜토리얼에서는 두 배열의 요소별 합을 계산하여 별도의 새로운 배열에 저장하는 프로그램을 작성해 보겠습니다.

두 개의 배열이 주어지면, 서로 같은 인덱스에 위치한 요소들을 더해야 합니다. 만약 그 합이 한 자릿수가 아니라면, 해당 숫자에서 각 자릿수를 추출하여 새 배열에 하나씩 저장합니다.

예시를 통해 문제를 이해해 보겠습니다.

입력

arr_one = {1, 2, 32, 4, 5}
arr_two = {1, 52, 3}

출력

2 5 4 3 5 4 5

문제 해결 절차

  • 더미 데이터로 두 개의 배열을 초기화합니다.

  • 새 배열의 크기를 미리 알 수 없으므로, 결과를 저장하기 위해 vector를 사용합니다.

  • 인덱스가 첫 번째 배열과 두 번째 배열의 길이보다 작은 동안 두 배열을 함께 순회합니다.

  • 같은 인덱스의 요소를 더한 뒤, pushDigits 함수를 통해 각 자릿수를 결과 벡터에 저장합니다.

  • 위 반복이 완료되면, 남아 있는 요소들에 대해 각 배열을 개별적으로 순회하며 자릿수를 추가합니다.

  • 최종적으로 벡터에 담긴 모든 요소를 출력합니다.

예제 코드

실제 동작하는 전체 코드를 살펴보겠습니다.

#include <bits/stdc++.h>
using namespace std;
void pushDigits(int num, vector<int> &result) {
   if (num > 0) {
      pushDigits(num / 10, result);
      result.push_back(num % 10);
   }
}
void printVector(vector<int> &result) {
   for (int i : result) {
      cout << i << " ";
   }
   cout << endl;
}
void addTwoArrayElements(vector<int> arr_one, vector<int> arr_two) {
   vector<int> result;
   int arr_one_length = arr_one.size(), arr_two_length = arr_two.size();
   int i = 0;
   while (i < arr_one_length && i < arr_two_length) {
      pushDigits(arr_one[i] + arr_two[i], result);
      i++;
   }
   while (i < arr_one_length) {
      pushDigits(arr_one[i++], result);
   }
   while (i < arr_two_length) {
      pushDigits(arr_two[i++], result);
   }
   printVector(result);
}
int main() {
   vector<int> arr_one = {1, 2, 32, 4, 5};
   vector<int> arr_two = {1, 52, 3};
   addTwoArrayElements(arr_one, arr_two);
   return 0;
}

실행 결과

위 프로그램을 실행하면 다음과 같은 결과를 얻을 수 있습니다.

2 5 4 3 5 4 5

동작 원리 설명

핵심 함수인 pushDigits는 재귀 호출을 활용해 숫자의 자릿수를 올바른 순서로 분리합니다. 먼저 num / 10으로 재귀 호출하여 앞자리부터 처리하고, 재귀가 반환된 후 num % 10으로 마지막 자릿수를 벡터에 추가하는 방식입니다. 덕분에 32 + 3처럼 두 자릿수 합이 발생하더라도 '3', '5' 순서대로 자연스럽게 저장됩니다.

또한 세 개의 while 루프를 사용해 두 배열의 길이가 서로 다른 경우에도 남은 요소들을 누락 없이 처리할 수 있습니다.

결론

이번 튜토리얼에서는 C++를 사용해 두 배열의 요소별 합을 구하고, 그 결과를 자릿수 단위로 새 배열에 저장하는 방법을 알아보았습니다. 진행 과정에서 궁금한 점이 있다면 댓글로 남겨 주세요.