Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++에서 합이 K로 나눌 수 있는 처음 N개의 자연수에서 쌍의 수

<시간/>

숫자 N과 K가 주어지면 합이 K로 나누어 떨어지는 쌍의 수를 계산해야 합니다. 예를 보겠습니다.

입력

N = 3
K = 2

출력

1

합이 K로 나누어 떨어지는 쌍은 단 하나뿐입니다. 그리고 쌍은 (1, 3)입니다.

알고리즘

  • N과 K를 초기화합니다.
  • N까지 자연수를 생성하여 배열에 저장합니다.
  • 카운트를 0으로 초기화합니다.
  • 배열에서 모든 쌍을 가져오려면 두 개의 루프를 작성하십시오.
    • 모든 쌍의 합을 계산합니다.
    • 쌍 합계가 K로 나누어지면 개수를 증가시킵니다.
  • 카운트를 반환합니다.

구현

다음은 위의 알고리즘을 C++로 구현한 것입니다.

#include <bits/stdc++.h>
using namespace std;
int get2PowersCount(vector<int> arr, int N, int K) {
   int count = 0;
   for (int i = 0; i < N; i++) {
      for (int j = i + 1; j < N; j++) {
         int sum = arr[i] + arr[j];
         if (sum % K == 0) {
            count++;
         }
      }
   }
   return count;
}
int main() {
   vector<int> arr;
   int N = 10, K = 5;
   for (int i = 1; i <= N; i++) {
      arr.push_back(i);
   }
   cout << get2PowersCount(arr, N, K) << endl;
   return 0;
}

출력

위의 코드를 실행하면 다음과 같은 결과를 얻을 수 있습니다.

9