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) по памяти, что эффективно для данной задачи.