leetcode1：两数之和

题目链接

leetcode

题目描述

给定一个整数数组 nums 和一个整数目标值 target，在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。
假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

C++ 代码

#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;

/*
    暴力枚举
    两重循环枚举下标 i、j，判断 nums[i] + nums[j] 是否等于 target。

    时间复杂度：O(n^2)
    空间复杂度：O(1)
*/
class Solution_0 {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res;
        for (int i = 0; i < nums.size(); i++) {
            for (int j = 0; j < i; j++) {
                if (nums[i] + nums[j] == target) {
                    res = vector<int> {j, i};
                    break;
                }
            }
            if (res.size() > 0) {
                break;
            }
        }
        return res;
    }
};

/*
    哈希表
    
    遍历一次数组，每次循环：
        判断 target − nums[i] 是否在哈希表
        将 nums[i] 插入哈希表
    
    时间复杂度：O(n)
    空间复杂度：O(n)
*/
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> hash;
        for (int i = 0; i < nums.size(); i++) {
            int other = target - nums[i];
            if (hash.count(other)) {
                return { hash[other], i };
            }
            hash[nums[i]] = i;
        }
        return {};
    }
};

// 辅助函数：打印数组
void printArray(const vector<int>& nums) {
    cout << "[";
    for (size_t i = 0; i < nums.size(); i++) {
        cout << nums[i];
        if (i != nums.size() - 1) cout << ",";
    }
    cout << "]";
}

int main() {
    Solution solution;
    vector<vector<int>> nums_cases = {
        {2,7,11,15},
        {3,2,4},    
        {3,3}
    };
    vector<int> target_cases = {9, 6, 6};

    for (int i = 0; i < nums_cases.size(); i++) {
        vector<int> nums = nums_cases[i];
        int target = target_cases[i];
        vector<int> indices = solution.twoSum(nums, target);

        cout << "Input: ";
        printArray(nums);
        cout << ", Target: " << target << endl;
        cout << "Output: ";
        printArray(indices);
        cout << endl;
    }
    
    return 0;
}