chore: update driver code (#1265)

codes/ruby/chapter_array_and_linkedlist/array.rb

@@ -76,32 +76,33 @@ end
 
 
 ### Driver Code ###
+if __FILE__ == $0
+  # 初始化数组
+  arr = Array.new(5, 0)
+  puts "数组 arr = #{arr}"
+  nums = [1, 3, 2, 5, 4]
+  puts "数组 nums = #{nums}"
 
-# 初始化数组
-arr = Array.new(5, 0)
-puts "数组 arr = #{arr}"
-nums = [1, 3, 2, 5, 4]
-puts "数组 nums = #{nums}"
+  # 随机访问
+  random_num = random_access(nums)
+  puts "在 nums 中获取随机元素 #{random_num}"
 
-# 随机访问
-random_num = random_access(nums)
-puts "在 nums 中获取随机元素 #{random_num}"
+  # 长度扩展
+  nums = extend(nums, 3)
+  puts "将数组长度扩展至 8 ，得到 nums = #{nums}"
 
-# 长度扩展
-nums = extend(nums, 3)
-puts "将数组长度扩展至 8 ，得到 nums = #{nums}"
+  # 插入元素
+  insert(nums, 6, 3)
+  puts "在索引 3 处插入数字 6 ，得到 nums = #{nums}"
 
-# 插入元素
-insert(nums, 6, 3)
-puts "在索引 3 处插入数字 6 ，得到 nums = #{nums}"
+  # 删除元素
+  remove(nums, 2)
+  puts "删除索引 2 处的元素，得到 nums = #{nums}"
 
-# 删除元素
-remove(nums, 2)
-puts "删除索引 2 处的元素，得到 nums = #{nums}"
+  # 遍历数组
+  traverse(nums)
 
-# 遍历数组
-traverse(nums)
-
-# 查找元素
-index = find(nums, 3)
-puts "在 nums 中查找元素 3 ，得到索引 = #{index}"
+  # 查找元素
+  index = find(nums, 3)
+  puts "在 nums 中查找元素 3 ，得到索引 = #{index}"
+end

codes/ruby/chapter_array_and_linkedlist/linked_list.rb

@@ -48,35 +48,36 @@ def find(head, target)
 end
 
 ### Driver Code ###
+if __FILE__ == $0
+  # 初始化链表
+  # 初始化各个节点
+  n0 = ListNode.new(1)
+  n1 = ListNode.new(3)
+  n2 = ListNode.new(2)
+  n3 = ListNode.new(5)
+  n4 = ListNode.new(4)
+  # 构建节点之间的引用
+  n0.next = n1
+  n1.next = n2
+  n2.next = n3
+  n3.next = n4
+  puts "初始化的链表为"
+  print_linked_list(n0)
 
-# 初始化链表
-# 初始化各个节点
-n0 = ListNode.new(1)
-n1 = ListNode.new(3)
-n2 = ListNode.new(2)
-n3 = ListNode.new(5)
-n4 = ListNode.new(4)
-# 构建节点之间的引用
-n0.next = n1
-n1.next = n2
-n2.next = n3
-n3.next = n4
-puts "初始化的链表为"
-print_linked_list(n0)
+  # 插入节点
+  insert(n0, ListNode.new(0))
+  print_linked_list n0
 
-# 插入节点
-insert(n0, ListNode.new(0))
-print_linked_list n0
+  # 删除节点
+  remove(n0)
+  puts "删除节点后的链表为"
+  print_linked_list(n0)
 
-# 删除节点
-remove(n0)
-puts "删除节点后的链表为"
-print_linked_list(n0)
+  # 访问节点
+  node = access(n0, 3)
+  puts "链表中索引 3 处的节点的值 = #{node.val}"
 
-# 访问节点
-node = access(n0, 3)
-puts "链表中索引 3 处的节点的值 = #{node.val}"
-
-# 查找节点
-index = find(n0, 2)
-puts "链表中值为 2 的节点的索引 = #{index}"
+  # 查找节点
+  index = find(n0, 2)
+  puts "链表中值为 2 的节点的索引 = #{index}"
+end

codes/ruby/chapter_array_and_linkedlist/list.rb

@@ -5,55 +5,56 @@ Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
 =end
 
 ### Driver Code ###
+if __FILE__ == $0
+  # 初始化列表
+  nums = [1, 3, 2, 5, 4]
+  puts "列表 nums = #{nums}"
 
-# 初始化列表
-nums = [1, 3, 2, 5, 4]
-puts "列表 nums = #{nums}"
+  # 访问元素
+  num = nums[1]
+  puts "访问索引 1 处的元素，得到 num = #{num}"
 
-# 访问元素
-num = nums[1]
-puts "访问索引 1 处的元素，得到 num = #{num}"
+  # 更新元素
+  nums[1] = 0
+  puts "将索引 1 处的元素更新为 0 ，得到 nums = #{nums}"
 
-# 更新元素
-nums[1] = 0
-puts "将索引 1 处的元素更新为 0 ，得到 nums = #{nums}"
+  # 清空列表
+  nums.clear
+  puts "清空列表后 nums = #{nums}"
 
-# 清空列表
-nums.clear
-puts "清空列表后 nums = #{nums}"
+  # 在尾部添加元素
+  nums << 1
+  nums << 3
+  nums << 2
+  nums << 5
+  nums << 4
+  puts "添加元素后 nums = #{nums}"
 
-# 在尾部添加元素
-nums << 1
-nums << 3
-nums << 2
-nums << 5
-nums << 4
-puts "添加元素后 nums = #{nums}"
+  # 在中间插入元素
+  nums.insert(3, 6)
+  puts "在索引 3 处插入元素 6 ，得到 nums = #{nums}"
 
-# 在中间插入元素
-nums.insert(3, 6)
-puts "在索引 3 处插入元素 6 ，得到 nums = #{nums}"
+  # 删除元素
+  nums.delete_at(3)
+  puts "删除索引 3 处的元素，得到 nums = #{nums}"
 
-# 删除元素
-nums.delete_at(3)
-puts "删除索引 3 处的元素，得到 nums = #{nums}"
+  # 通过索引遍历列表
+  count = 0
+  for i in 0...nums.length
+    count += nums[i]
+  end
 
-# 通过索引遍历列表
-count = 0
-for i in 0...nums.length
-  count += nums[i]
+  # 直接遍历列表元素
+  count = 0
+  nums.each do |x|
+    count += x
+  end
+
+  # 拼接两个列表
+  nums1 = [6, 8, 7, 10, 9]
+  nums += nums1
+  puts "将列表 nums1 拼接到 nums 之后，得到 nums = #{nums}"
+
+  nums = nums.sort { |a, b| a <=> b }
+  puts "排序列表后 nums = #{nums}"
 end
-
-# 直接遍历列表元素
-count = 0
-nums.each do |x|
-  count += x
-end
-
-# 拼接两个列表
-nums1 = [6, 8, 7, 10, 9]
-nums += nums1
-puts "将列表 nums1 拼接到 nums 之后，得到 nums = #{nums}"
-
-nums = nums.sort { |a, b| a <=> b }
-puts "排序列表后 nums = #{nums}"

codes/ruby/chapter_array_and_linkedlist/my_list.rb

@@ -35,7 +35,7 @@ class MyList
     # 元素数量超出容量时，触发扩容机制
     extend_capacity if size == capacity
     @arr[size] = num
-    
+
     # 更新元素数量
     @size += 1
   end
@@ -49,7 +49,7 @@ class MyList
 
     # 将索引 index 以及之后的元素都向后移动一位
     for j in (size - 1).downto(index)
-      @arr[j + 1] = @arr[j] 
+      @arr[j + 1] = @arr[j]
     end
     @arr[index] = num
 
@@ -95,37 +95,38 @@ class MyList
 end
 
 ### Driver Code ###
+if __FILE__ == $0
+  # 初始化列表
+  nums = MyList.new
 
-# 初始化列表
-nums = MyList.new
+  # 在尾部添加元素
+  nums.add(1)
+  nums.add(3)
+  nums.add(2)
+  nums.add(5)
+  nums.add(4)
+  puts "列表 nums = #{nums.to_array} ，容量 = #{nums.capacity} ，长度 = #{nums.size}"
 
-# 在尾部添加元素
-nums.add(1)
-nums.add(3)
-nums.add(2)
-nums.add(5)
-nums.add(4)
-puts "列表 nums = #{nums.to_array} ，容量 = #{nums.capacity} ，长度 = #{nums.size}"
+  # 在中间插入元素
+  nums.insert(3, 6)
+  puts "在索引 3 处插入数字 6 ，得到 nums = #{nums.to_array}"
 
-# 在中间插入元素
-nums.insert(3, 6)
-puts "在索引 3 处插入数字 6 ，得到 nums = #{nums.to_array}"
+  # 删除元素
+  nums.remove(3)
+  puts "删除索引 3 的元素，得到 nums = #{nums.to_array}"
 
-# 删除元素
-nums.remove(3)
-puts "删除索引 3 的元素，得到 nums = #{nums.to_array}"
+  # 访问元素
+  num = nums.get(1)
+  puts "访问索引 1 处的元素，得到 num = #{num}"
 
-# 访问元素
-num = nums.get(1)
-puts "访问索引 1 处的元素，得到 num = #{num}"
+  # 更新元素
+  nums.set(1, 0)
+  puts "将索引 1 处的元素更新为 0 ，得到 nums = #{nums.to_array}"
 
-# 更新元素
-nums.set(1, 0)
-puts "将索引 1 处的元素更新为 0 ，得到 nums = #{nums.to_array}"
-
-# 测试扩容机制
-for i in 0...10
-  # 在 i = 5 时，列表长度将超出列表容量，此时触发扩容机制
-  nums.add(i)
+  # 测试扩容机制
+  for i in 0...10
+    # 在 i = 5 时，列表长度将超出列表容量，此时触发扩容机制
+    nums.add(i)
+  end
+  puts "扩容后的列表 nums = #{nums.to_array} ，容量 = #{nums.capacity} ，长度 = #{nums.size}"
 end
-puts "扩容后的列表 nums = #{nums.to_array} ，容量 = #{nums.capacity} ，长度 = #{nums.size}"