Senior
Taakvoorwaarde: Gegeven een array van gehele getallen nums en een geheel getal target. Vind twee getallen in de array waarvan de som gelijk is aan target, en geef hun indices terug. Er wordt gegarandeerd dat er precies één oplossing bestaat, en dat hetzelfde element niet twee keer gebruikt kan worden. Voorbeelden: Invoer: nums = [2,7,11,15], target = 9 Uitvoer: [0,1]. Uitleg: nums[0] + nums[1] = 2 + 7 = 9 Invoer: nums = [3,2,4], target = 6 Uitvoer: [1,2]. Uitleg: nums[1] + nums[2] = 2 + 4 = 6 Invoer: nums = [3,3], target = 6 Uitvoer: [0,1]. Uitleg: nums[0] + nums[1] = 3 + 3 = 6
sobes.tech AI
Antwoord van 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) по памяти, что эффективно для данной задачи.