Видаліть записи з масиву, щоб сортувати його та збільшити масу елементів


13

Цей виклик - від іспиту на вступ до закритого курсу з кібербезпеки. У будь-якому разі це не має відношення до кібербезпеки, це лише перевірити логічність та навички кодування студентів.

Завдання

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

Вхід і вихід

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

Виходом буде низхідне відсортоване підмасив вхідного, сума якого більша, ніж будь-який інший можливий низхідний сортований підмасив.

Примітка: [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]

Пов'язані. (Я не можу перевірити, чи є два алгоритми насправді еквівалентними, але я думаю, що вони можуть бути.)
Мартін Ендер

Коментарі не для розширеного обговорення; ця розмова була переміщена до чату .
Мартін Ендер

Відповіді:


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

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

{-# 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. з'єднати ліву частину, новий підмасив та решту правої частини.

О(журналн)


1

Ця відповідь поширюється на тону Євангелія.

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

Т(i)=аi+макс[{0}{Т(j)|0j<iаiаj}]

(аi)Т(i)iТ

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()));
}

Спробуйте в Інтернеті!

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.