如何用一个函数返回链表头

要用一个函数返回链表头，可以按照以下步骤：

1. 定义一个结构体来表示链表的每个节点，包含数据和指向下一个节点的指针。

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

2. 定义一个函数来创建链表，并返回链表头。该函数接受一组数据作为输入，然后将数据依次插入到链表中。

ListNode* createLinkedList(vector<int> arr) {
    ListNode* head = NULL;
    ListNode* tail = NULL;
    for (int i = 0; i < arr.size(); i++) {
        ListNode* node = new ListNode(arr[i]);
        if (head == NULL) {
            head = node;
            tail = node;
        } else {
            tail->next = node;
            tail = node;
        }
    }
    return head;
}

3. 在主函数中调用 createLinkedList 函数来创建链表，并将返回结果赋值给指针变量 head。

int main() {
    vector<int> arr = {1, 2, 3, 4, 5};
    ListNode* head = createLinkedList(arr);
    // do something with the linked list
    return 0;
}