Удалить записи из массива, чтобы отсортировать их и максимизировать сумму элементов

13

Эта задача - от вступительного испытания до курса по кибербезопасности с закрытым числом. Во всяком случае, это не имеет отношения к кибербезопасности, это просто для проверки учащихся логических навыков и навыков кодирования.

задача

Напишите программу, которая удаляет записи из массива так, чтобы оставшиеся значения сортировались в строго убывающем порядке, а их сумма была максимизирована среди всех других возможных убывающих последовательностей.

Вход и выход

Входной будет массив целочисленных значений строго больше , чем 0и все отличаются друг от друга . Вы можете выбрать, читать ли входные данные из файла, командной строки или стандартного ввода.

Выход будет отсортированным по убыванию подмассивом входного массива, сумма которого больше, чем у любого другого возможного отсортированного по убыванию подмассива.

Примечание: [5, 4, 3, 2] является подмассивом [5, 4, 1, 3, 2], даже если 4и 3не являются смежными. Просто потому, что 1лопнуло.

Брутфорс решение

Самым простым решением, конечно, будет итерация среди всех возможных комбинаций данного массива и поиск отсортированного с наибольшей суммой, которая была бы в Python :

import itertools

def best_sum_desc_subarray(ary):
    best_sum_so_far = 0
    best_subarray_so_far = []
    for k in range(1, len(ary)):
        for comb in itertools.combinations(ary, k):
            if sum(comb) > best_sum_so_far and all(comb[j] > comb[j+1] for j in range(len(comb)-1)):
                best_subarray_so_far = list(comb)
                best_sum_so_far = sum(comb)
    return best_subarray_so_far

К сожалению, поскольку проверка того, отсортирован ли массив, и вычисление суммы его элементов, и так как эта операция будет выполняться раз для от до , асимптотическая сложность времени будет

Вызов

Ваша цель состоит в том, чтобы добиться лучшей сложности во времени, чем описанная выше грубая сила. Решение с наименьшей асимптотической сложностью по времени является победителем задачи. Если два решения имеют одинаковую асимптотическую сложность по времени, победителем будет решение с наименьшей асимптотической пространственной сложностью.

Примечание: вы можете рассмотреть чтение, запись и сравнение атомарных даже в больших количествах.

Примечание. Если имеется два или более решений, верните одно из них.

Контрольные примеры

Input:  [200, 100, 400]
Output: [400]

Input:  [4, 3, 2, 1, 5]
Output: [4, 3, 2, 1]

Input:  [50, 40, 30, 20, 10]
Output: [50, 40, 30, 20, 10]

Input:  [389, 207, 155, 300, 299, 170, 158, 65]
Output: [389, 300, 299, 170, 158, 65]

Input:  [19, 20, 2, 18, 13, 14, 8, 9, 4, 6, 16, 1, 15, 12, 3, 7, 17, 5, 10, 11]
Output: [20, 18, 16, 15, 12, 7, 5]

Input:  [14, 12, 24, 21, 6, 10, 19, 1, 5, 8, 17, 7, 9, 15, 23, 20, 25, 11, 13, 4, 3, 22, 18, 2, 16]
Output: [24, 21, 19, 17, 15, 13, 4, 3, 2]

Input:  [25, 15, 3, 6, 24, 30, 23, 7, 1, 10, 16, 29, 12, 13, 22, 8, 17, 14, 20, 11, 9, 18, 28, 21, 26, 27, 4, 2, 19, 5]
Output: [25, 24, 23, 22, 17, 14, 11, 9, 4, 2]
Marco
источник
Связанный. (Я не могу проверить прямо сейчас, являются ли эти два алгоритма фактически эквивалентными, но я думаю, что они могли бы быть.)
Мартин Эндер
Комментарии не для расширенного обсуждения; этот разговор был перенесен в чат .
Мартин Эндер

Ответы:

3

Perl

Это должно быть O (n ^ 2) во времени и O (n) в пространстве

Дать числа, разделенные пробелом в одной строке, в STDIN

#!/usr/bin/perl -a
use strict;
use warnings;

# use Data::Dumper;
use constant {
    INFINITY => 9**9**9,
    DEBUG    => 0,
};

# Recover sequence from the 'how' linked list
sub how {
    my @z;
    for (my $h = shift->{how}; $h; $h = $h->[1]) {
        push @z, $h->[0];
    }
    pop @z;
    return join " ", reverse @z;
}

use constant MINIMUM => {
    how  => [-INFINITY, [INFINITY]],
    sum  => -INFINITY,
    next => undef,
};

# Candidates is a linked list of subsequences under consideration
# A given final element will only appear once in the list of candidates
# in combination with the best sum that can be achieved with that final element
# The list of candidates is reverse sorted by final element
my $candidates = {
    # 'how' will represent the sequence that adds up to the given sum as a
    # reversed lisp style list.
    # so e.g. "1, 5, 8" will be represented as [8, [5, [1, INFINITY]]]
    # So the final element will be at the front of 'how'
    how  => [INFINITY],
    # The highest sum that can be reached with any subsequence with the same
    # final element
    sum  => 0,
    # 'next' points to the next candidate
    next => MINIMUM,   # Dummy terminator to simplify program logic
};

for my $num (@F) {
    # Among the candidates on which an extension with $num is valid
    # find the highest sum
    my $max_sum = MINIMUM;
    my $c = \$candidates;
    while ($num < $$c->{how}[0]) {
        if ($$c->{sum} > $max_sum->{sum}) {
            $max_sum = $$c;
            $c = \$$c->{next};
        } else {
            # Remove pointless candidate
            $$c = $$c->{next};
        }
    }

    my $new_sum = $max_sum->{sum} + $num;
    if ($$c->{how}[0] != $num) {
        # Insert a new candidate with a never before seen end element
        # Due to the unique element rule this branch will always be taken
        $$c = { next => $$c };
    } elsif ($new_sum <= $$c->{sum}) {
        # An already known end element but the sum is no improvement
        next;
    }
    $$c->{sum} = $new_sum;
    $$c->{how} = [$num, $max_sum->{how}];
    # print(Dumper($candidates));
    if (DEBUG) {
        print "Adding $num\n";
        for (my $c = $candidates; $c; $c = $c->{next}) {
            printf "sum(%s) = %s\n", how($c), $c->{sum};
        }
        print "------\n";
    }
}

# Find the sequence with the highest sum among the candidates
my $max_sum = MINIMUM;
for (my $c = $candidates; $c; $c = $c->{next}) {
    $max_sum = $c if $c->{sum} > $max_sum->{sum};
}

# And finally print the result
print how($max_sum), "\n";
Тон Хоспел
источник
3

О(NжурналN)О(N)

{-# LANGUAGE MultiParamTypeClasses #-}

import qualified Data.FingerTree as F

data S = S
  { sSum :: Int
  , sArr :: [Int]
  } deriving (Show)

instance Monoid S where
  mempty = S 0 []
  mappend _ s = s

instance F.Measured S S where
  measure = id

bestSubarrays :: [Int] -> F.FingerTree S S
bestSubarrays [] = F.empty
bestSubarrays (x:xs) = left F.>< sNew F.<| right'
  where
    (left, right) = F.split (\s -> sArr s > [x]) (bestSubarrays xs)
    sLeft = F.measure left
    sNew = S (x + sSum sLeft) (x : sArr sLeft)
    right' = F.dropUntil (\s -> sSum s > sSum sNew) right

bestSubarray :: [Int] -> [Int]
bestSubarray = sArr . F.measure . bestSubarrays

Как это устроено

bestSubarrays xsявляется последовательностью подмассивов, xsкоторые находятся на эффективной границе {наибольшая сумма, наименьший первый элемент}, упорядоченные слева направо путем увеличения суммы и увеличения первого элемента.

Чтобы перейти от bestSubarrays xsк bestSubarrays (x:xs), мы

  1. разбить последовательность на левую сторону с первыми элементами меньше x, и правую сторону с первыми элементами больше, чем x,
  2. найти новый подмассив, добавив xк крайнему правому подмассиву слева,
  3. отбросьте префикс подмассивов с правой стороны с меньшей суммой, чем у нового подмассива,
  4. объединить левую сторону, новый подмассив и оставшуюся часть правой стороны.

О(журналN)

Андерс Касеорг
источник
1

Этот ответ расширяет ответ Тон Хоспель.

Проблема может быть решена с помощью динамического программирования с использованием рекурсии

T(я)знак равноaя+Максимум[{0}{T(J)|0J<яaяaJ}]

(aя)T(я)яT

fn solve(arr: &[usize]) -> Vec<usize> {
    let mut tbl = Vec::new();
    // Compute table with maximum sums of any valid sequence ending
    // with a given index i.
    for i in 0..arr.len() {
        let max = (0..i)
            .filter(|&j| arr[j] >= arr[i])
            .map(|j| tbl[j])
            .max()
            .unwrap_or(0);
        tbl.push(max + arr[i]);
    }
    // Reconstruct an optimal sequence.
    let mut sum = tbl.iter().max().unwrap_or(&0).clone();
    let mut limit = 0;
    let mut result = Vec::new();

    for i in (0..arr.len()).rev() {
        if tbl[i] == sum && arr[i] >= limit {
            limit = arr[i];
            sum -= arr[i];
            result.push(arr[i]);
        }
    }
    assert_eq!(sum, 0);
    result.reverse();
    result
}

fn read_input() -> Vec<usize> {
    use std::io::{Read, stdin};
    let mut s = String::new();
    stdin().read_to_string(&mut s).unwrap();
    s.split(|c: char| !c.is_numeric())
        .filter(|&s| !s.is_empty())
        .map(|s| s.parse().unwrap())
        .collect()
}

fn main() {
    println!("{:?}", solve(&read_input()));
}

Попробуйте онлайн!

politza
источник