【算法练习】字符串处理 百练poj2944 单词替换

题目链接：http://bailian.openjudge.cn/practice/2944

2944:单词替换

2944:单词替换

总时间限制: 

1000ms

内存限制: 

65536kB

描述

输入一个字符串，以回车结束（字符串长度<=100）。该字符串由若干个单词组成，单词之间用一个空格隔开，所有单词区分大小写。现需要将其中的某个单词替换成另一个单词，并输出替换之后的字符串。

输入

输入包括3行，
第1行是包含多个单词的字符串 s;
第2行是待替换的单词a(长度 <= 100);
第3行是a将被替换的单词b(长度 <= 100).

s, a, b 最前面和最后面都没有空格.

输出

输出只有 1 行，将s中所有单词a替换成b之后的字符串。

样例输入

You want someone to help you
You
I

样例输出

I want someone to help you

---

题目理解：

主要就是处理输入

我用前面单词的第二种方法，cin.peek()

成功AC

AC代码如下：

#include<iostream>
#include <vector>
#include <string>
using namespace std;

vector<string> str;
string s;
int main(){
    //输入带空格的单词
    string strA,strB;  //A变成B
    while(cin>>s){
        str.push_back(s);  //输入空格分割的单词并且装入vector

        if(cin.peek()=='\n'){
            break;
        }
    }

    cin>>strA;
    cin>>strB;
    for(int i=0;i<str.size();i++){
        if(i==0){
            if(str[i]==strA)  cout<<strB;
            else cout<<str[i];
        }
        else{
            if(str[i]==strA)  cout<<" "<<strB;
            else cout<<" "<<str[i];
        }
    }
    return 0;
}