美文网首页
数组中的两数之和为指定的值,求此两数的索引

数组中的两数之和为指定的值,求此两数的索引

作者: 我才是臭吉吉 | 来源:发表于2019-02-21 00:35 被阅读1次

    题目

    给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

    你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。

    • 示例:
    给定 nums = [2, 7, 11, 15], target = 9
    因为 nums[0] + nums[1] = 2 + 7 = 9
    所以返回 [0, 1]
    

    解法1:暴力法

    依次遍历数组中的每个数据,查看与数组中的其他数据之和是否为指定的值。

    NSArray<NSNumber *> *getSumIndex1(NSArray<NSNumber *> *nums, NSInteger target) {
        // 注意:标准解法需要双重遍历原数组确定索引,时间复杂度为O(n^2),这里使用NSArray的API进行了简化
        NSArray *sumIndexInfo;
        for (NSNumber *item in nums) {
            // 当前索引
            NSInteger currentIndex = [nums indexOfObject:item];
            // 获取差值
            NSNumber *result = @(target - item.integerValue);
            // 查看是否在数组中
            NSInteger resultIndex = [nums indexOfObject:result];
            if ((resultIndex != NSNotFound)
                && (resultIndex != currentIndex)) {
                // 找到,得到索引并返回
                sumIndexInfo = @[@(currentIndex), @(resultIndex)];
                break;
            }
        }
        return sumIndexInfo;
    }
    

    解法2:两次遍历哈希表

    这里使用了NSDictionary类进行实现,目的是为了快速查找到数据对应的索引:

    NSArray<NSNumber *> *getSumIndex2(NSArray<NSNumber *> *nums, NSInteger target) {
        // 使用哈希表进行存取,加快查找速度
        NSMutableDictionary *hashTable = [@{} mutableCopy];
        for (NSInteger index = 0; index < nums.count; index++) {
            // key为数据值,value为索引(目的为了快速通过数值查询到索引)
            [hashTable setObject:@(index) forKey:nums[index]];
        }
    
        NSArray *sumIndexInfo;
        for (NSInteger index = 0; index < nums.count; index++) {
            NSInteger currentValue = [nums[index] integerValue];
            // 获取差值
            NSNumber *result = @(target - currentValue);
            // 在哈希表中查找对应索引
            NSNumber *resultIndex = [hashTable objectForKey:result];
            if ((resultIndex)
                && (resultIndex.integerValue != index)) {
                sumIndexInfo = @[@(index), resultIndex];
                break;
            }
        }
        return sumIndexInfo;
    }
    

    解法3:一次遍历哈希表

    遍历插入哈希表的同时,直接在表中查询,符合直接返回。

    NSArray<NSNumber *> *getSumIndex3(NSArray<NSNumber *> *nums, NSInteger target) {
        // 简化了方法2,一次遍历同时直接查询
        
        NSArray *sumIndexInfo;
        // 使用哈希表进行存取,加快查找速度
        NSMutableDictionary *hashTable = [@{} mutableCopy];
        for (NSInteger index = 0; index < nums.count; index++) {
            NSNumber *currentItem = nums[index];
            // 获取差值
            NSNumber *result = @(target - currentItem.integerValue);
            // 直接查询当前哈希表中是否已经包含差值对象
            NSNumber *resultIndex = [hashTable objectForKey:result];
            if ((resultIndex)
                && (resultIndex.integerValue != index)) {
                sumIndexInfo = @[@(index), resultIndex];
                break;
            } else {
                // key为数据值,value为索引(目的为了快速通过数值查询到索引)
                [hashTable setObject:@(index) forKey:currentItem];
            }
        }
        return sumIndexInfo;
    }
    

    由于在OC的数组中已经包含了根据元素直接返回索引的高级API,故大幅度简化了解法1,使用哈希表快速返回索引也变得没有那么必要了。

    相关文章

      网友评论

          本文标题:数组中的两数之和为指定的值,求此两数的索引

          本文链接:https://www.haomeiwen.com/subject/glrlyqtx.html