一个文件,两列,多行。
第一列是字母,第二列是数字,同列数据之间通过空格分割。
统计首列字母相同的第二列之和。
样例输入:
A 5
B 6
OO 7
A 6
A 2
OO 2
输出:
A:13
B:6
OO:9
面试官要求30分钟内实现,python,C++实现。
夸下5分钟写完一种语言海口的我花了18分钟睡觉……(晕)
然后开始整理思路,一开始写了很久vector+查找算法的(因为考虑map插入会导致覆盖),写了半天发现……太复杂了,而且查找算法确实应用的不熟练。后来转变一下思路map可以取出原来的值再加进去。。我去
捡起来半生不熟的map用法。。
思路就是:整个数据看作key-value结构,map的特性就是如果找一个不存在的,会自动插入并赋值默认值。value求和只需要找出对应key的map元素,取出当前的value(如果没有,int之类的默认是0),和新value相加后再塞回去。python直接利用字典这个结构即可。
#include<map>
#include<string>
#include<iostream>
#include<fstream>
#include<sstream> using namespace std; typedef double datatype; template<typename T>
void Stringtoother(string str, T &result)//编写的模板函数供之后重用,实现字符串到任意类型(如果可以)的转换
{
istringstream is(str);
is >> result;
} int main(int argc, char** argv) {
map<string, datatype> mp;//声明最终存储的map ifstream in("D:\\test.txt");//测试样例,完全符合要求的情况下测试正常(未考虑文件出错的情况) //ifstream in(argv[1]);//实际的路径,可以通过cmd读取 if (in.fail())
{
cout << "文件打开失败!" << endl;
system("pause");
return -;
} string s;//文件读取每一行存储到的字符串
mp.clear();
while (getline(in, s))
{
int tag = s.find(" ");
string temp1, temp2;
temp1 = s.substr(, tag - );
temp2 = s.substr(tag + , s.size()); int value = ;
Stringtoother(temp2, value);
mp[temp1] += value;//键值先读取原来的,然后再相加后覆盖 } map<string, datatype>::iterator it;//最终结果输出
for (it = mp.begin(); it != mp.end(); it++) {
cout << it->first << ":" << it->second << endl;
} system("pause");
return ;
}
python实现
def readFile(path):
dict_data={}
with open(path,'r') as df:
for kv in [d.strip().split(' ') for d in df]:
temp=dict_data.get(kv[0],0)
dict_data[kv[0]]=float(kv[1])+temp
return dict_data; test_data={}
test_data=readFile("D:\\test.txt")
print(test_data)
凉凉