вторник, 24 декабря 2013 г.

Codility. Train. Perm-Missing-Elem ★★

A zero-indexed array A consisting of N different integers is given. The array contains integers in the range [1..(N + 1)], which means that exactly one element is missing.
Your goal is to find that missing element.
Write a function:
def solution(A)
that, given a zero-indexed array A, returns the value of the missing element.
For example, given array A such that:
  A[0] = 2
  A[1] = 3
  A[2] = 1
  A[3] = 5
the function should return 4, as it is the missing element.
Assume that:
  • N is an integer within the range [0..100,000];
  • the elements of A are all distinct;
  • each element of array A is an integer within the range [1..(N + 1)].
Complexity:
  • expected worst-case time complexity is O(N);
  • expected worst-case space complexity is O(1), beyond input storage (not counting the storage required for input arguments).
Elements of input arrays can be modified.


Solution
def solution(A):
    j = 1
    A.sort()
    for i in A:
        if i != j:
            break
        j = j + 1
    return j

6 комментариев:

  1. def solution(A):
    return sum(xrange(1,len(A)+2)) - sum(A)

    ОтветитьУдалить
    Ответы
    1. Этот комментарий был удален автором.

      Удалить
    2. Шалфей, если не трудно, объясните пожалуйста логику, почему именно len(A)+2?

      Спасибо

      Удалить
    3. жевать здесь особо нечего. Мы знаем что у нас последовательность от 1 до N+1, но один элемент пропущен и поэтому элементов в массиве N. Идея такая - получить полный список этих элементов до N+1, а так как xrange(1, N) вернет нам {1,2,3, ... , N-1}, то его приходится выравнивать прибавляя 2. как-то так.

      хотя я сейчас уже не стал бы так делать :) :
      def solution(A):
      return (lambda lm: ((1 + lm) * lm / 2))(len(A)+1) - sum(A)

      так чуть длиннее писать, зато экономнее считать :)

      Удалить
    4. Этот комментарий был удален автором.

      Удалить
    5. вот придумал еще один вариант.
      основная идея все та же, а вот реализация чуток новая:
      https://codility.com/demo/results/demoXNW5GF-WSW/

      Удалить