Question

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* removeElements(struct ListNode* head, int val) {

    struct ListNode *beforeNode = head;

    while(beforeNode != NULL){

        if(head == beforeNode && head->val == val){
            struct ListNode* q = head;
            head = q->next;
            beforeNode = head;
            free(q);
        }
        else if(beforeNode->next != NULL && beforeNode->next->val == val){
            struct ListNode *p = beforeNode->next;
            beforeNode = p->next;
            free(p);
        }

       else
           beforeNode = beforeNode->next;

    }

    return head;
}

Answer 1

此代码在Xcode上有效，但在leetcode编译器中失败

好吧，该代码在任何平台上均不起作用。

您在Xcode上进行的测试必须不完整，因为该代码在任何平台上均具有未定义的行为。

看这个简单的例子：

#include <stdio.h>
#include <stdlib.h>

struct ListNode
{
  struct ListNode *next;
  int val;
};

void printList(struct ListNode* p)
{
  if (p)
  {
    printf("%p %d", (void*)p, p->val);
    while(p->next)
    {
      printf("->");
      p = p->next;
      printf("%p %d", (void*)p, p->val);
    }
  }
  printf("\n");
}


// Function from question
struct ListNode* removeElements(struct ListNode* head, int val) {

    struct ListNode *beforeNode = head;

    while(beforeNode != NULL){

        if(head == beforeNode && head->val == val){
            struct ListNode* q = head;
            head = q->next;
            beforeNode = head;
            free(q);
        }
        else if(beforeNode->next != NULL && beforeNode->next->val == val){
            struct ListNode *p = beforeNode->next;
            beforeNode = p->next;
            printf("Free val %p %d\n", (void*)p, p->val);
            free(p);
        }

       else
           beforeNode = beforeNode->next;

    }

    return head;
}

int main()
{
  // Initialize a list with three element like: 1->42->1->NULL
  struct ListNode *head = malloc(sizeof *head);
  head->val = 1;
  head->next = malloc(sizeof *head);
  head->next->val = 42;
  head->next->next = malloc(sizeof *head);
  head->next->next->val = 1;
  head->next->next->next = NULL;

  printList(head);

  removeElements(head, 42);

  printList(head);

  return 0;
}

示例输出：

0x558311c02260 1->0x558311c02280 42->0x558311c022a0 1
Free val 0x558311c02280 42
0x558311c02260 1->0x558311c02280 42

如您所见，存在两个问题：

结果列表为1-> 42，但我们希望为1-> 1，换句话说，该列表已损坏。
最后一行中打印的节点是我们刚刚释放的节点（即0x558311c02280）。那是未定义的行为。

问题是这一行：

beforeNode = p->next;

应该是

beforeNode->next = p->next;

上述更改后的输出：

0x561def3e6260 1->0x561def3e6280 42->0x561def3e62a0 1
Free val 0x561def3e6280 42
0x561def3e6260 1->0x561def3e62a0 1

现在列表是正确的，没有使用释放的内存。

此代码在Xcode上有效，但在leetcode编译器中失败

1 个答案: