Senior
36
Условие задачи: Дан массив целых чисел nums и целое число target. Нужно найти два числа в массиве, которые в сумме дают target, и вернуть их индексы. Гарантируется, что решение существует ровно одно, и нельзя использовать один и тот же элемент дважды. Примеры: Input: nums = [2,7,11,15], target = 9 Output: [0,1]. Объяснение: nums[0] + nums[1] = 2 + 7 = 9 Input: nums = [3,2,4], target = 6 Output: [1,2]. Объяснение: nums[1] + nums[2] = 2 + 4 = 6 Input: nums = [3,3], target = 6 Output: [0,1]. Объяснение: nums[0] + nums[1] = 3 + 3 = 6
Компании, где спрашивали
Ответ от нейросети
sobes.tech 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) по памяти, что эффективно для данной задачи.