展开或不展开时仅重新加载表视图的部分- Swift

import UIKit

class ViewController: UIViewController, UITableViewDataSource, UITableViewDelegate {
    // 假设有3个节
    let numberOfSections = 3
    // 每个节的行数
    let numberOfRows = [3, 4, 5]
    // 记录每个节是否展开
    var isExpanded = [false, false, false]
    
    @IBOutlet weak var tableView: UITableView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        tableView.dataSource = self
        tableView.delegate = self
    }
    
    // UITableViewDataSource协议方法，返回节的数量
    func numberOfSections(in tableView: UITableView) -> Int {
        return numberOfSections
    }
    
    // UITableViewDataSource协议方法，返回每个节的行数
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return isExpanded[section] ? numberOfRows[section] : 0
    }
    
    // UITableViewDataSource协议方法，返回每个单元格的内容
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)
        cell.textLabel?.text = "Section \(indexPath.section), Row \(indexPath.row)"
        return cell
    }
    
    // UITableViewDelegate协议方法，处理节的头部点击事件
    func tableView(_ tableView: UITableView, didSelectHeaderInSection section: Int) {
        isExpanded[section] = !isExpanded[section]
        
        // 重新加载被点击的节
        tableView.reloadSections(IndexSet(integer: section), with: .automatic)
    }
}