单词替换OpenJudge1.7编程基础之字符串-21

#include<iostream>
#include<string>
using namespace std;
int main()
{
	string s, a, b, word;
	int h = 0;
	getline(cin, s);
	cin >> a;
	cin >> b;
	s += ' ';
	for (int i = 0;i < s.size();i++)
	{
		if (s[i] == ' ')  //开始写的时候把"=="写成了"="，一直发现不了，到了第二天再看就发现了
		{
			word = s.substr(h, i - h);
			if ((word.compare(a)) == 0)
			{
				cout << b;
			}
			else
			{
				cout << word;
			}
			if (i != s.size() - 1)
			{
				cout << " ";
			}
			h = i+1;
		}
	}
	return 0;
}