Циклический сдвиг связанного списка вправо
Условие задачи
Дана голова односвязного списка и число k. Выполните циклический сдвиг списка вправо на k узлов.
Решение
Циклический сдвиг можно описать и по-другому: взять подсписок из последних k узлов связанного списка и присоединить его к началу. Кроме того, нужно выполнить ещё три действия:
- Соединить последний узел связанного списка с головой, поскольку после сдвига у списка будет другой хвост.
- Новой головой связанного списка станет узел в начале переносимого подсписка.
- Узел непосредственно перед началом переносимого подсписка станет новым хвостом сдвинутого списка.
Код
Вот как будет выглядеть наш алгоритм:
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
61
62
63
64
65
66
67
68
69
70
package main
import "fmt"
type ListNode struct {
Value int
Next *ListNode
}
func rotate(head *ListNode, k int) *ListNode {
if head == nil || head.Next == nil || k <= 0 {
return head
}
lastNode := head
listLength := 1
for lastNode.Next != nil {
lastNode = lastNode.Next
listLength++
}
rotations := k % listLength
if rotations == 0 {
return head
}
lastNode.Next = head
nodesToSkip := listLength - rotations
newTail := head
for i := 0; i < nodesToSkip-1; i++ {
newTail = newTail.Next
}
newHead := newTail.Next
newTail.Next = nil
return newHead
}
func newList(values ...int) *ListNode {
dummy := &ListNode{}
current := dummy
for _, value := range values {
current.Next = &ListNode{Value: value}
current = current.Next
}
return dummy.Next
}
func printList(head *ListNode) {
first := true
for current := head; current != nil; current = current.Next {
if !first {
fmt.Print(" ")
}
fmt.Print(current.Value)
first = false
}
fmt.Println()
}
func main() {
head := newList(1, 2, 3, 4, 5, 6)
head = rotate(head, 3)
fmt.Print("Узлы сдвинутого связанного списка: ")
printList(head)
}
Вывод:
1
Узлы сдвинутого связанного списка: 4 5 6 1 2 3
Временная сложность
Временная сложность алгоритма равна \(O(N)\), где \(N\) — общее количество узлов в связанном списке.
Пространственная сложность
Мы использовали только постоянный объём памяти, поэтому пространственная сложность алгоритма равна \(O(1)\).