-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path61. Rotate List.txt
60 lines (51 loc) · 2.22 KB
/
61. Rotate List.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
//Просто совет: всякий раз, когда вы видите смену позиций типа вращения, помните, что это шаблон, и его можно использовать для оптимизации вашего кода. В этом случае вам не
//нужно каждый раз считать k вращений, вам просто нужно рассчитать k%n вращений, где n — количество узлов. пройти весь связанный список один раз лучше, чем выполнить k
//поворотов, когда k очень велико.
//Оптимальный код. Проходим все узлы списка, замыкаем конечный узел на первый, ищем новую голову и хвост.
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(!head) return head;
int len=1; // количество узлов
ListNode *newH, *tail;
newH=tail=head;
while(tail->next) // получить количество узлов в списке
{
tail = tail->next;
len++;
}
tail->next = head; // замыкаем последний узел списка на первый узел списка
if(k %= len)
{
for(auto i=0; i<len-k; i++) tail = tail->next; // Хвостовой узел - это (len-k)-й узел (1-й узел - головной)
}
newH = tail->next; //Новое начало списка = следущему списку, после конечного элемента
tail->next = NULL; //В последнем элементе объевляем NULL в next
return newH;
}
};
//Мой код. 3 цикла. 1-ый добавляет все узлы в вектор, 2-ой выставляет их в правельный порядок, 3-ий связывает узлы в новый порядок.
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if (head == NULL) return head;
vector<ListNode*> allNodes;
ListNode *temp = head;
while (temp != NULL) {
allNodes.push_back(temp);
temp = temp->next;
}
int startPos = k % allNodes.size(); //Подсчёт на какой позиции окажется первый узел.
for (int i = 0; i < startPos; i++){
allNodes.insert(allNodes.begin(), allNodes.back());
allNodes.erase(allNodes.begin() + allNodes.size() - 1);
}
temp = allNodes[0];
for (int i = 1; i < allNodes.size(); i++) {
temp->next = allNodes[i];
temp = temp->next;
}
temp->next = NULL;
return allNodes[0];
}
};