给c++写python的split()与input()【python一样写c++、一】

python的split确实是很香的功能。

写c++的时候，就会想着，要是能直接input().split()那不挺好。

实际上真的可以：自己动手，丰衣足食。

先放成品展示。

int main(){
    auto k=input().split();//k的类型是vector<string>
	cout<<k<<endl;
}

完整代码在文章末尾。

1.首先，写一个input()不难，getline+return就行

mystring input(string c=""){
	cout<<c;
	mystring s;
	getline(cin,s);
	return s;
}

为什么是mystring呢？因为我们想要实现input().split()的效果。普通的string不能胜任，所以自己构造一个mystring的类（具体往下看）

2.写一个split()函数。

考虑到便捷性，直接把它写成成员函数。

这个mystring类就继承string类。写成class mystring:public string也行。

用struct主要是应为加入的成员函数都是public。

实现部分就爆扫过去，扫到ch（默认空格）就记录位置，然后插入到一个vector<string>里

struct mystring:string{//继承string类，写成class mystring：public string也行
    mystring() :string() {}
    mystring& operator=(const string &c) {
        (*this).assign(c);
        return *this;
    }
    vector<string> split(char c=' '){
        vector<string> x;
        int last=0,len=0;
        for(int i=0;i<size();++i){
            if(at(i)==c){
                x.push_back(substr(last,len));
                last=i+1;
                len=0;
            }
            else{
                len++;
            }
        }
        if(last<size())
            x.push_back(substr(last,len));
        return x;
    }
}s;

其实更简洁与接近python的是使用find，但时间复杂度都是O(n)，写好了我也不想改了。

本文作者XXOY

3.OK！

我们就可以直接

auto k=input().split();

用auto是因为简单，也可以写成vector<string> k;

4.加一个输出函数

由于c++不能直接打印vector，所以我们自己写一个重载。

具体的写法下次再教（可以关注我），这次先放个例子。

template<typename T>
ostream& operator<<(ostream &o,vector<T> &x){
    char ch=0;
    if(typeid(T)==typeid(string)) ch='\'';
    o<<"["<<ch<<x[0]<<ch;
    for(int i=1;i<x.size();++i){
        o<<", "<<ch<<x[i]<<ch;
    }
    o<<"]";
    return o;
}

还加入了string的特判，加一个‘，更接近python。

OK，完毕，下面是完整代码

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

struct mystring:string{//继承string类
    mystring() :string() {}
    mystring& operator=(const string &c) {
        (*this).assign(c);
        return *this;
    }
    vector<string> split(char c=' '){
        vector<string> x;
        int last=0,len=0;
        for(int i=0;i<size();++i)
            if(at(i)==c){
                x.push_back(substr(last,len));
                last=i+1,len=0;
            }
            else ++len;
        if(last<size())
            x.push_back(substr(last,len));
        return x;
    }
}s;

template<typename T>
ostream& operator<<(ostream &o,vector<T> &x){
    char ch=0;
    if(typeid(T)==typeid(string)) ch='\'';
    o<<"["<<ch<<x[0]<<ch;
    for(int i=1;i<x.size();++i){
        o<<", "<<ch<<x[i]<<ch;
    }
    o<<"]";
    return o;
}

mystring input(string c=""){
	cout<<c;
	mystring s;
	getline(cin,s);
	return s;
}

int main()
{
    auto k=input().split();//k的类型是vector<string>
	cout<<k<<endl;
}

//本文作者XXOY

这就是input()、split() 的全部内容了。

我还写了比如类似python map一样，转换列表（vector）的东西，将和那个cout<<vector一起，分成两篇来讲。

比如这样

    auto c=my_map(my_int,k);//k是vector<string>
    cout<<c;

后续请看我主页（应该能咕完）。

我是XXOY，我们下次再见。