NavigationController实现返回第一个控制器

在第一个控制器中添加一个按钮

let shutDownBtn: UIButton!

分别添加以下代码

    override func viewWillDisappear(_ animated: Bool) {
        super.viewWillDisappear(true)
        self.shutDownBtn.isHidden = true
    }

    override func viewWillAppear (_ animated: Bool) {
        super.viewWillAppear(true)
        shutDownBtn = UIButton.init()
        shutDownBtn.setTitle("关闭", for: .normal)
        shutDownBtn.setTitleColor(UIColor.black, for: .normal)
        //        shutDownBtn.setImage(UIImage.init(named: "compose_guide_icon_close_check"), for: .normal)
        shutDownBtn.frame = CGRect(x: self.view.bounds.width - 120, y: 0, width: 40, height: 40)

        shutDownBtn.addTarget(self, action: #selector(shutdown), for: .touchUpInside)

        self.navigationController?.navigationBar.addSubview(shutDownBtn)
        self.shutDownBtn.isHidden = false
    }