多线程依赖问题

遇到一个面试题：
3个线程A，B，C存在依赖关系，B依赖A执行结束，C依赖B执行结束，请设计实现。

具体实现如下：

首先，创建一个锁对象：

@Data
public class LockFlag {
    private int flag = 0;
}

线程A的实现代码

public class Ajob extends Thread {
    private LockFlag lockFlag;

    public Ajob(LockFlag lockFlag) {
        this.lockFlag = lockFlag;
    }

    @Override
    public void run() {
        synchronized (lockFlag) {
                try {
                    System.out.println("threadA starting ");
                    if (lockFlag.getFlag() > 0) {
                        lockFlag.notify();
                    } else {
                        lockFlag.setFlag(1);
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
}

线程B的实现：

public class Bjob  extends  Thread{

    private LockFlag lockFlag1;
    private LockFlag lockFlag2;

    public Bjob(LockFlag lockFlag1, LockFlag lockFlag2) {
        this.lockFlag1 = lockFlag1;
        this.lockFlag2 = lockFlag2;
    }

    @Override
    public void run() {
        synchronized (lockFlag1) {

            try {
                if (lockFlag1.getFlag() == 0) {
                    lockFlag1.setFlag(1);
                    lockFlag1.wait();
                }
                System.out.println("ThreadB starting");

                synchronized (lockFlag2) {
                    if (lockFlag2.getFlag() > 0) {
                        lockFlag2.notify();
                    } else {
                        lockFlag2.setFlag(1);
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}

线程C实现：

public class Cjob extends Thread {
    private LockFlag lockFlag2;

    public Cjob(LockFlag lockFlag2) {
        this.lockFlag2 = lockFlag2;
    }

    @Override
    public void run() {
        synchronized (lockFlag2) {
            try {
                if (lockFlag2.getFlag() == 0) {
                    lockFlag2.setFlag(1);
                    lockFlag2.wait();
                }
                System.out.println("ThreadC starting");
            } catch (Exception e){
                e.printStackTrace();
            }
        }
    }
}

main方法调用

public class TestMain {
    public static void main(String[] args) {

        LockFlag lockFlag1 = new LockFlag();
        LockFlag lockFlag2 = new LockFlag();

        new Ajob(lockFlag1).start();
        new Bjob(lockFlag1, lockFlag2).start();
        new Cjob(lockFlag2).start();
    }
}