C ++在Vector中分组重复

时间:2017-09-12 07:27:29

标签: c++ vector grouping repeat

我的文件结构如下:

A 123456 0
G 123456 5
A 235334 0
B 123456 2

每条信息的存储方式如下:

temp.code >> temp.personid >> temp.data

我已将此信息存储在Vector

 ifstream fin("test.txt");
 vector<TestClass> test;
 TestClass temp;
 string line;
 while (getline(fin, line)) {//.. test.push_back(temp);}

给定的 personid 可以在文件中多次出现。 我想要做的是迭代向量并将重复分组到每个 personid 的单个类对象中,我的目标是我想要为每个特定对象求和,以便输出上面的文件将是:

123456 : 7
235334 : 0

接近这个会有什么优雅的方式?

由于

2 个答案:

答案 0 :(得分:1)

以下代码使用评论建议的std::unordered_map。它将逐行读取您的文件 该代码假定该人的ID属于int类型,代码类型为std::string,数据类型为int

它将每个Person(此处为结构示例)插入到地图中。如果一个人的身份已经在那里,它将总结数据。这意味着此解决方案不使用临时std::vector,而只使用std::unordered_map

See live example with your data on ideone.com

<强>代码:

#include <iostream>
#include <sstream>
#include <fstream>
#include <string>
#include <unordered_map>

struct Person
{
   std::string code;
   int         data;
};

typedef std::unordered_map<int, Person> PersonMap;

int main()
{
   std::ifstream fin("test.txt");

   PersonMap persons;

   /* line by line reading */
   for (std::string line; std::getline(fin, line); )
   {
      std::istringstream iss(line);

      int    personId;
      Person personData;

      /* parse line as std::string, int, int */
      iss >> personData.code >> personId >> personData.data;

      /* insert into map and save result */
      std::pair<PersonMap::iterator, bool> insertResult =
         persons.insert(std::pair<int, Person>(personId, personData));

      /* if personId is already there */
      if (!insertResult.second)
      {
         insertResult.first->second.data += personData.data;
      }
   }

   /* output whole map */
   for(auto const &person : persons)
   {
      std::cout << person.first << " : " << person.second.data << "\n";
   }
   std::cout << std::flush;
}

<强>输出:

235334 : 0
123456 : 7

答案 1 :(得分:0)

使用Unordered map。无序地图中的查找时间在平均情况下始终为 O(1) 。我使用vector作为样本数据,你可以从文件加载数据而不是向量。

#include <bits/stdc++.h>
using namespace std;

int main() {
    unordered_map<string, int>m;
    unordered_map<string, int>::iterator itr;    // Iterator to iterate unordered map
    vector<pair<string, int> >person_details;    // pair of vector to represent sample data, you can load data from file instead
    person_details.push_back(make_pair("123456",0));
    person_details.push_back(make_pair("123456",5));
    person_details.push_back(make_pair("235334",0));
    person_details.push_back(make_pair("123456",2));
    for(int i=0;i<person_details.size();i++)
    {
        if(m.find(person_details[i].first) == m.end() )                // If personId is not present in map, insert it
            m[person_details[i].first]=person_details[i].second;
        else m[person_details[i].first]+=person_details[i].second;        // If personId is present in map, increment it.
    }
    for(itr=m.begin();itr!=m.end();itr++)          
        cout<<itr->first<<" "<<itr->second<<endl;       // Displaying personId with occurance
    return 0;
}

Output:
235334 0
123456 7

注意:您可以使用Map来保持 O(LogN) 时间,其中<​​strong> N 是容器的大小。