wzoi-单词替换

题目描述:

输入一个字符串，以回车结束（字符串长度<=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

时间限制: 1000ms
空间限制: 128MB

上代码：

#include <bits/stdc++.h>
using namespace std;
string s,a,b;
int main() {
	getline(cin,s);
	s+=" ";
	string w="";
	cin>>a>>b;
	for(int i=0; i<s.length(); i++) {
		if(s[i]!=' ') {
			w+=s[i];
		} else {
			if(w==a)cout<<b<<" ";
			else cout<<w<<" ";
			w="";
		}
	}
	return 0;
}