【Leetcode】 two sum #1 for rust solution
阿新 • • 發佈:2020-08-18
給定一個整數陣列 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 asi32); 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