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

C++ STL multiset count() 함수 – 사용법, 문법 및 예제 총정리

이 글에서는 C++ STL의 multiset::count() 함수가 어떻게 동작하는지, 그 문법과 실제 예제를 통해 자세히 알아보겠습니다.

C++ STL에서 multiset이란 무엇인가?

multiset(멀티셋)은 set 컨테이너와 매우 유사한 연관 컨테이너입니다. set과 마찬가지로 값을 키(key) 형태로 저장하며, 항상 정렬된 순서를 유지합니다.

multiset에서도 값 자체가 곧 키로 식별된다는 점은 set과 동일하지만, 두 컨테이너의 결정적인 차이는 다음과 같습니다.

  • set: 모든 키가 고유해야 하며, 중복된 키를 허용하지 않습니다.
  • multiset: 동일한 키 값이 여러 개 존재할 수 있습니다.

multiset은 내부적으로 균형 이진 탐색 트리(Binary Search Tree) 구조로 구현되어 있어, 삽입·삭제·검색이 O(log n)의 시간 복잡도로 수행됩니다.

multiset::count()란?

multiset::count()는 C++ STL에 내장된 함수로, <set> 헤더 파일에 정의되어 있습니다.

이 함수는 특정 키와 일치하는 요소의 개수를 세어 반환합니다. multiset은 같은 키 값을 여러 개 가질 수 있으므로, 특정 값이 컨테이너 안에 몇 번 등장하는지 알고 싶을 때 count()를 활용할 수 있습니다.

count()는 컨테이너 전체를 대상으로 해당 키를 검색하고 그 결과를 반환합니다. 만약 찾고자 하는 키가 컨테이너에 존재하지 않으면 0을 반환하므로, 요소의 존재 여부를 확인하는 용도로도 유용하게 사용됩니다.

문법(Syntax)

ms_name.count(value_type T);

매개변수(Parameters)

이 함수는 multiset의 값 타입(value_type)과 같은 타입의 인자 하나를 받습니다. 이 인자는 연관된 multiset 컨테이너 내에서 검색하고자 하는 값입니다.

반환 값(Return Value)

전달된 키와 동일한 값을 가지는 요소의 개수를 반환합니다. 시간 복잡도는 로그 시간에 해당 개수만큼 더해진 O(log n + count)입니다.

예제 1: 기본 동작 확인

입력: std::multiset<int> mymultiset = {1, 2, 2, 3, 2, 4};
    mymultiset.count(2);
출력: 3

위 예제에서 값 2는 컨테이너에 3번 저장되어 있으므로, count(2)는 3을 반환합니다.

예제 2: 전체 코드

#include <bits/stdc++.h>
using namespace std;
int main() {
    int arr[] = {1, 2, 3, 1, 1, 1};
    multiset<int> check(arr, arr + 6);
    cout<<"List is : ";
    for (auto i = check.begin(); i != check.end(); i++)
        cout << *i << " ";
    cout << "\n1 is occuring: "<<check.count(1)<<" times";
    return 0;
}

출력 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

List is : 1 1 1 1 2 3
1 is occuring 4 times

배열의 요소들이 multiset에 삽입되면 자동으로 오름차순으로 정렬되며, 값 1이 총 4개 존재함을 확인할 수 있습니다.

예제 3: 여러 키 카운트하기

#include <bits/stdc++.h>
using namespace std;
int main() {
    int arr[] = {1, 2, 3, 1, 1, 1, 2, 2};
    multiset<int> check(arr, arr + 8);
    cout<<"List is : ";
    for (auto i = check.begin(); i != check.end(); i++)
    cout << *i << " ";
    cout << "\n1 is occuring: "<<check.count(1)<<" times";
    cout << "\n2 is occuring: "<<check.count(2)<<" times";
    return 0;
}

출력 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

List is : 1 1 1 1 2 2 2 3
1 is occuring 4 times
2 is occuring 3 times

마무리

multiset::count()는 중복 값을 허용하는 multiset에서 특정 값의 빈도를 손쉽게 파악할 수 있는 간편하고 효율적인 도구입니다. 존재하지 않는 키에 대해서는 0을 반환하기 때문에 별도의 예외 처리 없이 안전하게 사용할 수 있다는 점도 큰 장점입니다.