LeetCode | Rotate List

最新推荐文章于 2024-10-14 18:52:56 发布

益达915

最新推荐文章于 2024-10-14 18:52:56 发布

阅读量236

点赞数

分类专栏： LeetCode 算法链表文章标签： leetcode 链表环

本文链接：https://blog.csdn.net/u013033845/article/details/52080754

版权

算法同时被 3 个专栏收录

114 篇文章 0 订阅

订阅专栏

LeetCode

100 篇文章 0 订阅

订阅专栏

链表

13 篇文章 0 订阅

订阅专栏

一开始想错了，以为要在k的位置旋转
其实是要循环滚动地推k次，比如
[1,2]
2
结果应当为[1,2]，以为推一圈又回来了
而
[1,2]
3
则应当是[2,1]效果和1是一样的

于是加一个mod判断就好了


class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        //空或左侧没有链
        if(head==NULL || head->next==NULL) return head;
        ListNode* new_head;
        ListNode root(head->val-1);
        root.next=head;

        ListNode* t=head;
        int count=0;
        while(t!=NULL){
            count++;
            t=t->next;
        }
        //找出要推多少次
        k%=count;
        if(k==0) return head;
        k=count-k;

        ListNode* pre=&root;
        t=head;
        for(count=0;t!=NULL;pre=pre->next,t=t->next,count++){
            if(count==k){
                new_head=t;
                // //获取最后一个节点
                ListNode* temp=t;
                while(temp->next!=NULL) temp=temp->next;

                //连接尾部到头部
                temp->next=root.next;

                pre->next=NULL;
                break;
            }
        }
        return new_head;
    }
};

其实代码可以优雅一点

      //空或左侧没有链
      if(head==NULL || head->next==NULL) return head;
      ListNode* new_head;

      ListNode* t=head;
      int count=1;
      while(t->next!=NULL){
          count++;
          t=t->next;
      }
      //找出要推多少次
      k%=count;
      if(k==0) return head;
      k=count-k;

      //连接尾部到头,在前面遍历到最后的时候，就可以保留这个指针
      t->next=head;

      for(count=0;t->next!=NULL;t=t->next,count++){
          if(count==k){
              new_head=t->next;
              t->next=NULL;
              break;
          }
      }
      return new_head;