在刷算法题的时候总是会遇到需要处理格式化输入的问题,这里记录下将string类型的数据,格式化输入到vector数组中,提供两种方法:
1、C语言中的strtok。
因为strtok为不安全函数,这里使用strtok_r.
直接看代码:
example:
void formatStringTovector()
{
string line;
getline(cin, line);
char *S = (char *)line.c_str();
char *context = NULL;
char *target = strtok_r(S, ",", &context);
vector<int> arr;
while(target != NULL) {
int num = stoi(target);
arr.push_back(num);
target = strtok_r(NULL, "," , &context);
}
int total = arr.back();
arr.pop_back();
cout << "tatal : " << total << endl;
for(auto x : arr) {
cout << x << " ";
}
cout << endl;
}
- strtok_r :
函数原型:
char *strtok_r(char *str, const char *delim, char **saveptr);
参数:
char *str :输入的字符串
const char *delim : 分割字符串的依据(以什么作为分割点)
char **savptr : 存储分割后剩余的字符串
返回值:
返回值是个指针,用于返回从被分割字符串中得到的第一个目标子串
注意:
strtok_r 第一次调用第一个参数传入字符串,第二次调用传入NULL
另外第一个参数要求char *类型,上面使用了string中的c_str()方法,将string类型转为char *类型。
改为",," 以两个逗号为分隔符也行,输出如下:
二、stringstream与getline
example2:
void formatStringTovector2()
{
string line;
getline(cin, line);
stringstream ss(line);
char delim = ',';
string tmpString;
vector<int> arr;
while (getline(ss, tmpString, delim)) {
int num = stoi(tmpString);
arr.push_back(num);
}
int total = arr.back();
arr.pop_back();
cout << "tatal : " << total << endl;
for(auto x : arr) {
cout << x << " ";
}
cout << endl;
return;
}
- stringstream
说明:stringstream是C++提供的串流(stream)东西。作用就是从string对象读取字符或字符串。
这里是将输入的字符串作为stringstream的输入。
- getline
函数原型:std::istream& getline(std::istream& is, std::string& str, char delim);
参数说明:std::istream & is : 输入流对象,可以是
std::cin
(标准输入)、文件流对象(std::ifstream
)、字符串流对象(std::istringstream
)等等。std::string & str : 传入参数,存储读取到的一行文本的字符串,也就是被分割出去的那个字符串。
char deelim:指定分割字符。
注意: strtok_r 指定的分割可以是字符串,为const char *类型;getline不行,只能为一个字符 。