C++实现数组元素和给定值相等的下标数组并返回vector

直接上代码

#include <iostream>
#include <vector>

#define CNT 32
using namespace std;

vector<int> equal_vector(int arr[], int item, int cnt)
{
    vector<int> ivec={};
    for (int i = 0; i < cnt; ++i) {
        if(arr[i] == item){
            ivec.push_back(i);
        }
    }
    return ivec;
}

int main()
{
    int zhenjishu[CNT]={1,2,3,3,1,2,1};
    int present_num=7;
    vector<int> Zhend[CNT]={{}};
    for (int i=0;i<present_num;i++)
    {
        int zhen = zhenjishu[i];
        Zhend[i]=equal_vector(zhenjishu,zhen,present_num);
	int count = Zhend[i].size();
	for (int j = 0; j < count;j++)
	{
	    cout << Zhend[i][j] << " ";
	}
	cout << endl;
    }
   cout << "Hello World";
   return 0;
}

运行结果

0 4 6
1 5
2 3
2 3
0 4 6
1 5
0 4 6
Hello World