Senior
Condition de l'exercice: Étant donné un tableau d'entiers nums et un entier target. Il faut trouver deux nombres dans le tableau dont la somme est égale à target, et retourner leurs indices. Il est garanti qu'une seule solution existe, et qu'on ne peut pas utiliser le même élément deux fois. Exemples: Entrée : nums = [2,7,11,15], target = 9 Sortie : [0,1]. Explication : nums[0] + nums[1] = 2 + 7 = 9 Entrée : nums = [3,2,4], target = 6 Sortie : [1,2]. Explication : nums[1] + nums[2] = 2 + 4 = 6 Entrée : nums = [3,3], target = 6 Sortie : [0,1]. Explication : nums[0] + nums[1] = 3 + 3 = 6
sobes.tech IA
Réponse de l'IA
Для решения задачи "два числа в массиве, сумма которых равна 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) по памяти, что эффективно для данной задачи.