PTA-复数类的定义(java)

题目描述：

编写一个复数类，可以进行复数加法和减法运算。编写一个包含main方法的类测试该复数类。要求该复数类至少包含一个无参的构造方法和一个带参的构造方法；数据成员包括复数的实部和虚部，为double类型；包括两个方法，分别实现复数的加法和减法运算。测试代码如下：

    public static void main(String [] args){
            Complex a=new Complex();
            Complex b=new Complex();
            Scanner in=new Scanner(System.in);
            a.setRealPart(in.nextDouble());
            a.setImaginaryPart(in.nextDouble());
            b.setRealPart(in.nextDouble());
            b.setImaginaryPart(in.nextDouble());
            System.out.println(a);
            System.out.println(b);
            System.out.println(a.add(b));
            System.out.println(a.sub(b));      
}

输入格式：

输入两个复数，输入为两行，每一行为一个复数实部和虚部，用空格隔开

输出格式：

输出复数加法和减法结果。输出为4行，第一行和第二行输出两个复数，第三行为两个复数的加法运算结果，第四行为减法运算结果。

整体代码如下：

import java.util.Scanner;
class Complex{
    private double real;
    private double imag;

    public Complex() {
    }

    public Complex(double real, double imag) {
        this.real = real;
        this.imag = imag;
    }

    public double getReal() {
        return real;
    }

    public void setRealPart(double real) {
        this.real = real;
    }

    public double getImag() {
        return imag;
    }

    public void setImaginaryPart(double imag) {
        this.imag = imag;
    }
    public String add(Complex b){
        double r = this.real+b.real;
        double i = this.imag+b.imag;
        String res = "";
        if (r==0&&i!=0){
            res += i+"i";
        }else if (i==0&&r!=0){
            res += r;
        } else if (this.real==0&&this.imag==0){
            res +="0.0";
        } else{
            if (i>0){
                res += r+"+"+i+"i";
            }
            else {
                res += r+i+"i";
            }
        }
        return  res;
    }
    public String sub(Complex b){
       double r = this.real-b.real;
       double i =this.imag-b.imag;
        String res = "";
        if (r==0&&i!=0){
            res += i+"i";
        }else if (i==0&&r!=0){
            res += r;
        } else if (this.real==0&&this.imag==0){
            res +="0.0";
        } else{
            if (i>0){
                res += r+"+"+i+"i";
            }
            else {
                res += r+""+i+"i";
            }
        }
        return  res;
    }

    @Override
    public String toString() {
        String res ="";
        if (this.real==0&&this.imag!=0){
            res += this.imag+"i";
        }else if (this.imag==0&&this.real!=0){
            res += this.real;
        } else if (this.real==0&&this.imag==0){
            res +="0.0";
        } else{
            if (this.imag>0){
                res += this.real+"+"+this.imag+"i";
            }
            else {
                res += this.real+this.imag+"i";
            }
        }
        return  res;
    }
}
public class Main{
    public static void main(String [] args){
        Complex a=new Complex();
        Complex b=new Complex();
        Scanner in=new Scanner(System.in);
        a.setRealPart(in.nextDouble());
        a.setImaginaryPart(in.nextDouble());
        b.setRealPart(in.nextDouble());
        b.setImaginaryPart(in.nextDouble());
        System.out.println(a);
        System.out.println(b);
       System.out.println(a.add(b));
      System.out.println(a.sub(b));
    }
}