Rimuovere le voci dall'array per ordinarlo e massimizzare la somma degli elementi


13

Questa sfida va da un test di ammissione a un corso di cyber security a numero chiuso. Comunque non ha a che fare con la sicurezza informatica, è solo per testare le abilità logiche e di programmazione degli studenti.

Compito

Scrivi un programma che rimuove le voci da un array in modo che i valori rimanenti siano ordinati in un ordine strettamente decrescente e la loro somma sia massimizzata tra tutte le altre possibili sequenze decrescenti.

Ingresso e uscita

L'input sarà un array di valori interi rigorosamente maggiori di 0e tutti diversi l'uno dall'altro . Sei libero di scegliere se leggere l'input dal file, dalla riga di comando o dallo stdin.

L'output sarà in ordine decrescente subarray in di quello in input, la cui somma è maggiore di qualsiasi altro subarray in ordine decrescente.

Nota: [5, 4, 3, 2] è un sottoarray di [5, 4, 1, 3, 2], anche se 4e 3non sono adiacenti. Solo perché è 1stato fatto scoppiare.

Soluzione Bruteforce

La soluzione più semplice, naturalmente, sarebbe iterare tra tutte le possibili combinazioni dell'array dato e cercarne una ordinata con la somma maggiore, che sarebbe, in 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

Sfortunatamente, poiché controllando se l'array è ordinato e calcolando la somma dei suoi elementi è e poiché questa operazione verrà eseguita volte per da a , la complessità del tempo asintotico sarà

Sfida

Il tuo obiettivo è ottenere una migliore complessità temporale rispetto alla forza bruta di cui sopra. La soluzione con la più piccola complessità temporale asintotica è la vincitrice della sfida. Se due soluzioni hanno la stessa complessità temporale asintotica, il vincitore sarà quello con la minore complessità spaziale asintotica.

Nota: puoi prendere in considerazione la lettura, la scrittura e il confronto atomico anche su grandi numeri.

Nota: se ci sono due o più soluzioni, restituire una di esse.

Casi test

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]

Relazionato. (Non posso verificare in questo momento se i due algoritmi sono effettivamente equivalenti, ma penso che potrebbero essere.)
Martin Ender,

I commenti non sono per una discussione estesa; questa conversazione è stata spostata in chat .
Martin Ender,

Risposte:


3

Perl

Dovrebbe essere O (n ^ 2) nel tempo e O (n) nello spazio

Assegna numeri separati da spazio su una riga a 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

O(nlogn)O(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

Come funziona

bestSubarrays xsè la sequenza di subarrays xsche si trovano sulla frontiera efficiente di {somma più grande, primo elemento più piccolo}, ordinata da sinistra a destra aumentando la somma e aumentando il primo elemento.

Per andare da bestSubarrays xsa bestSubarrays (x:xs), noi

  1. dividere la sequenza in un lato sinistro con i primi elementi minori di xe un lato destro con i primi elementi maggiori di x,
  2. trova un nuovo subarray anteponendo xil subarray più a destra sul lato sinistro,
  3. rilasciare il prefisso dei sottoarray dal lato destro con una somma inferiore rispetto al nuovo sottoarray,
  4. concatena il lato sinistro, il nuovo sottoarray e il resto del lato destro.

O(logn)


1

Questa risposta si espande su quella di Ton Hospel.

Il problema può essere risolto con la programmazione dinamica usando la ricorrenza

T(io)=un'io+max[{0}{T(j)|0j<ioun'ioun'j}]

dove (un'io) è la sequenza di input e T(io) la somma ottenibile al massimo di qualsiasi sotto-sequenza decrescente che termina con indice io. La soluzione effettiva può quindi essere rintracciata usandoT, come nel seguente codice ruggine.

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

Provalo online!

Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.