如何使用Swift将表视图数据按字母顺序划分为多个部分？(重写)

import UIKit

class ViewController: UIViewController, UITableViewDataSource {
    @IBOutlet weak var tableView: UITableView!
    
    var data = ["Apple", "Banana", "Cherry", "Durian", "Grape", "Kiwi", "Lemon", "Mango", "Orange", "Peach", "Pear", "Strawberry", "Watermelon"]
    var groupedData = [String: [String]]()
    var sectionTitles = [String]()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // Group data by first letter
        for item in data {
            let firstLetter = String(item.prefix(1))
            if var group = groupedData[firstLetter] {
                group.append(item)
                groupedData[firstLetter] = group
            } else {
                groupedData[firstLetter] = [item]
            }
        }
        
        // Sort section titles
        sectionTitles = groupedData.keys.sorted()
        
        tableView.dataSource = self
    }
    
    // MARK: - UITableViewDataSource
    
    func numberOfSections(in tableView: UITableView) -> Int {
        return sectionTitles.count
    }
    
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        let sectionTitle = sectionTitles[section]
        return groupedData[sectionTitle]?.count ?? 0
    }
    
    func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? {
        return sectionTitles[section]
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)
        
        let sectionTitle = sectionTitles[indexPath.section]
        if let items = groupedData[sectionTitle] {
            let item = items[indexPath.row]
            cell.textLabel?.text = item
        }
        
        return cell
    }
}