C++에서 합이 3의 배수인 크기 2 또는 3의 가능한 모든 그룹 계산

이 튜토리얼에서는 합이 3의 배수인 크기가 2 또는 3인 가능한 그룹의 수를 찾는 프로그램에 대해 논의할 것입니다.

이 튜토리얼에서는 합이 3의 배수인 크기가 2 또는 3인 가능한 그룹의 수를 찾는 프로그램에 대해 논의할 것입니다.

예시

#include<bits/stdc++.h>
using namespace std;
//returning count of pairs of
//2 or 3
int count_groups(int arr[], int n){
   int c[3] = {0}, i;
   int res = 0;
   for (i=0; i<n; i++)
      c[arr[i]%3]++;
   res += ((c[0]*(c[0]-1))>>1);
   res += c[1] * c[2];
   res += (c[0] * (c[0]-1) * (c[0]-2))/6;
   res += (c[1] * (c[1]-1) * (c[1]-2))/6;
   res += ((c[2]*(c[2]-1)*(c[2]-2))/6);
   res += c[0]*c[1]*c[2];
   return res;
}
int main(){
   int arr[] = {3, 6, 7, 2, 9};
   int n = sizeof(arr)/sizeof(arr[0]);
   cout << "Required number of groups are " << count_groups(arr,n) << endl;
   return 0;
}

출력

Required number of groups are 8