pta 7-4 h0084. 单词替换

题目

输入一个字符串，以回车结束（字符串长度不超过100）。

该字符串由若干个单词组成，单词之间用一个空格隔开，所有单词区分大小写。

现需要将其中的某个单词替换成另一个单词，并输出替换之后的字符串。

输入格式:

输入共3行。

第1行是包含多个单词的字符串 s;

第2行是待替换的单词a(长度不超过100);

第3行是a将被替换的单词b(长度不超过100)。

输出格式:

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

输入样例:

You want someone to help you
You
I

输出样例:

I want someone to help you

代码

java

//java
import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String s = sc.nextLine();    //输入3行数据
        String a = sc.nextLine();
        String b = sc.nextLine();
        s = replace(s,a,b);    //调用替换函数
        System.out.print(s);    //输出
    }
    public static String replace(String s,String a,String b){
        String[] s2 = s.split(" ");    //以空格分割s存s2
        for(int i=0;i<s2.length;i++){
            if(s2[i].equals(a)){    //逐个对比s2与a是否相同，相同则替换
                s2[i]=b;
            }
        }
        s="";    //清空s，因为后面要重新赋值
        for(String i:s2){
            s+=i+" ";    //将替换后的s2加上空格赋值回给s
        }
        return s;    //返回s
    }
}

c++解题代码有机会补上