1. 程式人生 > 實用技巧 >【Leetcode】 two sum #1 for rust solution

【Leetcode】 two sum #1 for rust solution

給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。

你可以假設每種輸入只會對應一個答案。但是,陣列中同一個元素不能使用兩遍。



示例:

給定 nums = [2, 7, 11, 15], target = 9

因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

Rust Solution:

 1     pub fn two_sum(nums: Vec<i32>, target: i32) -> Vec<i32> {
 2         let mut hashmap = HashMap::new
(); 3 let mut vec: Vec<i32> = Vec::new(); 4 for (value1, &item ) in nums.iter().enumerate() { 5 let temp = target - item; 6 if let Some(&value2) = hashmap.get(&temp) { 7 vec.push(value2 as i32); 8 vec.push(value1 as
i32); 9 } 10 hashmap.insert(item, value1); 11 } 12 vec 13 }

C++ Solution:

 1     vector<int> twoSum(vector<int>& nums, int target) {
 2         unordered_map<int, int> _map;
 3         for(int i = 0; i < nums.size(); i++)
 4         {
5 int temp = target - nums[i]; 6 if(_map.find(temp) != _map.end()) { 7 return { _map[temp], i}; 8 } 9 _map[nums[i]] = i; 10 } 11 return {}; 12 }

Python Solution:

1 def twoSum(self, nums: List[int], target: int) -> List[int]:
2         dict = {};
3         for i in range(len(nums)):
4             if (target - nums[i]) in dict :
5                 return [dict[target - nums[i]], i]
6             dict[nums[i]] = i

Github link : https://github.com/DaviRain-Su/leetcode_solution/tree/master/two_sum