std::vector 判斷vector容器中是否存在某元素

工作中經常遇見的一個場景：判斷某個元素是否在vector容器中。

當然，會有很多種方法，由內置數據類型到自定義數據類型，下面簡單總結一下。

【1】內置數據類型

代碼勝過一切文檔。如下示例代碼：

#include <iostream>
#include <vector>
#include <string>

// 為了便於示例，聲明全局容器
std::vector<std::string> strVec;

void methods(const std::string& target)
{
    // 方法一：遍歷容器，查找相等元素判斷是否存在
    {
        for (const auto& item : strVec)
        {
            if (item == target)
            {
                std::cout << "method1: find " << target << " exists." << std::endl;
                break;
            }
        }
    }
    // 方法二：獲取元素個數，通過個數判斷是否存在
    {
        int nCount = std::count(strVec.begin(), strVec.end(), target);
        if (nCount > 0)
        {
            std::cout << "method2: find " << target << " exists." << std::endl;
        }
    }
    // 方法三：查詢元素迭代器，通過迭代器有效性判斷是否存在
    {
        auto iter = std::find(strVec.begin(), strVec.end(), target);
        if (iter != strVec.end())
        {
            std::cout << "method3: find " << target << " exists." << std::endl;
        }
    }
    // 方法四：查詢相等元素的迭代器，通過迭代器有效性判斷是否存在
    {
        auto iter = std::find_if(strVec.begin(), strVec.end(), [&](const std::string& item)->bool
            { return (item == target); });
        if (iter != strVec.end())
        {
            std::cout << "method4: find " << target << " exists." << std::endl;
        }
    }
}

int main()
{
    strVec = { "C", "C++", "Java", "Python", "Lua", "Sql" };
    // 場景1：查找Ruby
    std::cout << "Find Ruby" << std::endl;
    methods("Ruby");
    // 場景2：查找C++
    std::cout << "Find C++" << std::endl;
    methods("C++");

    system("pause");
    return 0;
}

// result
/*
Find Ruby
Find C++
method1: find C++ exists.
method2: find C++ exists.
method3: find C++ exists.
method4: find C++ exists.
*/

【2】自定義數據類型

代碼勝過一切文檔。如下示例代碼：

#include <iostream>
#include <map>
#include <string>
#include <vector>

struct Student
{
    std::string code;  // 學號（唯一性標識）
    int grade;         // 年級
    std::map<std::string, double> scores; // <科目，成績>

    bool operator==(const Student& obj) const
    {
        return obj.code == code; // 只要學號相同即可
    }
};

// 為了便於示例，聲明全局容器
std::vector<Student> vecStu;

void methods(const Student& target)
{
    // 方法一：遍歷容器，查找相等元素判斷是否存在
    {
        for (const auto& item : vecStu)
        {
            if (item == target)
            {
                std::cout << "method1: find exists." << std::endl;
                break;
            }
        }
    }
    // 方法二：獲取元素個數，通過個數判斷是否存在
    {
        int nCount = std::count(vecStu.begin(), vecStu.end(), target);
        if (nCount > 0)
        {
            std::cout << "method2: find exists." << std::endl;
        }
    }
    // 方法三：查詢元素迭代器，通過迭代器有效性判斷是否存在
    {
        auto iter = std::find(vecStu.begin(), vecStu.end(), target);
        if (iter != vecStu.end())
        {
            std::cout << "method3: find exists." << std::endl;
        }
    }
    // 方法四：查詢相等元素的迭代器，通過迭代器有效性判斷是否存在
    {
        if (std::find_if(vecStu.begin(), vecStu.end(), [&](const Student& obj)->bool
            { return (obj == target); }) != vecStu.end())
        {
            std::cout << "method4: find exists." << std::endl;
        }
    }
}

int main()
{
    vecStu.push_back({ "080605109", 6, { {"English", 100}, {"China", 100} } });
    vecStu.push_back({ "080605110", 7, { {"English", 99}, {"China", 70} } });
    vecStu.push_back({ "080605111", 8, { {"English", 98}, {"China", 69} } });
    vecStu.push_back({ "080605112", 6, { {"English", 97}, {"China", 68} } });
    vecStu.push_back({ "080605113", 7, { {"English", 96}, {"China", 67} } });
    Student obj = { "080605114", 8, { {"English", 95}, {"China", 66} } };
    vecStu.push_back(obj);

    // 場景1：查找學號為[0806005109]的學生，我們暫不關注成績
    std::cout << "Find code: 0806005108" << std::endl;
    methods(Student{ "0806005108", {} });
    // 場景2：查找學號為[0806005114]的學生，我們暫不關注成績
    std::cout << "Find code: 0806005114" << std::endl;
    methods(obj);
    // 場景3：不想通過學號，只想查找六年級，是否存在英語和語文全為滿分的學生
    auto iter = std::find_if(vecStu.begin(), vecStu.end(), [&](const Student& obj)->bool
        { return (obj.grade == 6 && 
            obj.scores.find("English")->second == 100 &&
            obj.scores.find("China")->second == 100); });
    if (iter != vecStu.end())
    {
        std::cout << "method: find 100 exists." << std::endl;
    }

    return 0;
}

// result
/*
Find code: 0806005108
Find code: 0806005114
method1: find exists.
method2: find exists.
method3: find exists.
method4: find exists.
method: find 100 exists.
*/

注意：自定義數據類型，必須重載==符號。

原因很簡單：遇見這種場景，計算機弄不明白你想讓它以什么標准來判斷兩個對象相等，所以你得給它確定了相等的標准或准則。

綜上所述，再強調一點：針對自定義數據類型，使用std::find_if方法，顯而易見，自由度很大。

 

good good study, day day up.

順序 選擇 循環 總結