Sobes.tech
Senior

Podmínka úlohy: Dán je pole celých čísel nums a celé číslo target. Je třeba najít dvě čísla v poli, jejichž součet je roven target, a vrátit jejich indexy. Je zaručeno, že řešení existuje přesně jedno, a nelze použít stejný prvek dvakrát. Příklady: Vstup: nums = [2,7,11,15], target = 9 Výstup: [0,1]. Vysvětlení: nums[0] + nums[1] = 2 + 7 = 9 Vstup: nums = [3,2,4], target = 6 Výstup: [1,2]. Vysvětlení: nums[1] + nums[2] = 2 + 4 = 6 Vstup: nums = [3,3], target = 6 Výstup: [0,1]. Vysvětlení: nums[0] + nums[1] = 3 + 3 = 6

sobes.tech AI

Odpověď od AI

Для решения задачи "два числа в массиве, сумма которых равна target" можно использовать хэш-таблицу (Map) для хранения уже просмотренных чисел и их индексов. Идея в том, чтобы при обходе массива для каждого элемента проверить, есть ли в Map число, равное target минус текущий элемент. Если есть — возвращаем индексы.

Пример на Java:

import java.util.HashMap;
import java.util.Map;

public class TwoSum {
    public static int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];
            if (map.containsKey(complement)) {
                return new int[] {map.get(complement), i};
            }
            map.put(nums[i], i);
        }
        throw new IllegalArgumentException("No two sum solution");
    }

    public static void main(String[] args) {
        int[] nums = {2, 7, 11, 15};
        int target = 9;
        int[] result = twoSum(nums, target);
        System.out.println("[" + result[0] + ", " + result[1] + "]");
    }
}

Такой подход работает за O(n) по времени и O(n) по памяти, что эффективно для данной задачи.