hello-algo/en/codes/swift/chapter_searching/linear_search.swift

54 lines
1.4 KiB
Swift

/**
* File: linear_search.swift
* Created Time: 2023-01-28
* Author: nuomi1 (nuomi1@qq.com)
*/
import utils
/* Linear search (array) */
func linearSearchArray(nums: [Int], target: Int) -> Int {
// Traverse array
for i in nums.indices {
// Found the target element, thus return its index
if nums[i] == target {
return i
}
}
// Did not find the target element, thus return -1
return -1
}
/* Linear search (linked list) */
func linearSearchLinkedList(head: ListNode?, target: Int) -> ListNode? {
var head = head
// Traverse the list
while head != nil {
// Found the target node, return it
if head?.val == target {
return head
}
head = head?.next
}
// If the target node is not found, return null
return nil
}
@main
enum LinearSearch {
/* Driver Code */
static func main() {
let target = 3
/* Perform linear search in array */
let nums = [1, 5, 3, 2, 4, 7, 5, 9, 10, 8]
let index = linearSearchArray(nums: nums, target: target)
print("Index of target element 3 = \(index)")
/* Perform linear search in linked list */
let head = ListNode.arrToLinkedList(arr: nums)
let node = linearSearchLinkedList(head: head, target: target)
print("The corresponding node object for target node value 3 is \(node!)")
}
}