Sobes.tech
Senior

Ülesande tingimus: Antud on täisarvude massiiv nums ja täisarv target. Tuleb leida massiivist kaks arvu, mille summa võrdub targetiga, ning tagastada nende indeksid. Garanteeritakse, et lahendus on täpselt üks ning sama elementi ei saa kasutada kaks korda. Näited: Sisend: nums = [2,7,11,15], target = 9 Väljund: [0,1]. Selgitus: nums[0] + nums[1] = 2 + 7 = 9 Sisend: nums = [3,2,4], target = 6 Väljund: [1,2]. Selgitus: nums[1] + nums[2] = 2 + 4 = 6 Sisend: nums = [3,3], target = 6 Väljund: [0,1]. Selgitus: nums[0] + nums[1] = 3 + 3 = 6

sobes.tech AI

Vastus AI-lt

Для решения задачи "два числа в массиве, сумма которых равна 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) по памяти, что эффективно для данной задачи.