如果单元格被点击则显示按钮，离开后在UITableView中隐藏

import UIKit

class CustomTableViewCell: UITableViewCell {
    let button = UIButton(type: .system)
    
    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        
        // 设置按钮的属性
        button.setTitle("按钮", for: .normal)
        button.frame = CGRect(x: 0, y: 0, width: 60, height: 30)
        button.isHidden = true
        
        // 将按钮添加到单元格中
        addSubview(button)
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    override func layoutSubviews() {
        super.layoutSubviews()
        
        // 根据按钮的显示状态来调整其他视图的布局
        if button.isHidden {
            // 隐藏按钮时的布局
        } else {
            // 显示按钮时的布局
        }
    }
}

class ViewController: UIViewController, UITableViewDelegate, UITableViewDataSource {
    let tableView = UITableView()
    var selectedIndexPath: IndexPath?
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 设置UITableView的属性
        tableView.delegate = self
        tableView.dataSource = self
        tableView.frame = view.bounds
        view.addSubview(tableView)
    }
    
    // UITableViewDelegate协议方法：单元格被点击时调用
    func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        // 记录被点击的单元格的索引
        selectedIndexPath = indexPath
        
        // 刷新被点击的单元格
        tableView.reloadRows(at: [indexPath], with: .none)
    }
    
    // UITableViewDelegate协议方法：单元格将要显示时调用
    func tableView(_ tableView: UITableView, willDisplay cell: UITableViewCell, forRowAt indexPath: IndexPath) {
        // 判断是否需要显示按钮
        if indexPath == selectedIndexPath {
            // 显示按钮
            if let customCell = cell as? CustomTableViewCell {
                customCell.button.isHidden = false
            }
        } else {
            // 隐藏按钮
            if let customCell = cell as? CustomTableViewCell {
                customCell.button.isHidden = true
            }
        }
    }
    
    // UITableViewDataSource协议方法：返回单元格数量
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return 10
    }
    
    // UITableViewDataSource协议方法：返回单元格
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = CustomTableViewCell(style: .default, reuseIdentifier: nil)
        
        // 配置单元格的其他内容
        
        return cell
    }
}