Wyzwanie

Załóżmy, że masz listę liczb i wartość docelową. Znajdź zestaw wszystkich kombinacji swoich liczb, które składają się na wartość docelową, zwracając je jako indeksy listy.

Wejście i wyjście

Dane wejściowe przyjmą listę liczb (niekoniecznie unikalną) i docelową liczbę sumowania. Dane wyjściowe będą zbiorem niepustych list, z których każda zawiera wartości całkowite odpowiadające pozycji wartości na oryginalnej liście wejściowej.

Przykłady

Input: values = [1, 2, 1, 5], target = 8
Output: [ [0,1,3], [1,2,3] ]

Input: values = [4.8, 9.5, 2.7, 11.12, 10], target = 14.8
Output: [ [0,4] ]

Input: values = [7, 8, 9, -10, 20, 27], target = 17
Output: [ [1,2], [0,3,4], [3,5] ]

Input: values = [1, 2, 3], target = 7
Output: [ ]

Punktacja

To jest kod-golf , więc wygrywa najkrótszy kod!

Łuska , 10 bajtów

ηλfo=¹ṁ⁰tṖ

1-indeksowany. Wypróbuj online!

Wyjaśnienie

ηλfo=¹ṁ⁰tṖ  Inputs are a number n (explicit, accessed with ¹) and a list x (implicit).
η           Act on the incides of x
 λ          using this function:
         Ṗ   Take all subsets,
        t    remove the first one (the empty subset),
  f          and keep those that satisfy this:
      ṁ⁰      The sum of the corresponding elements of x
   o=¹        equals n.

Wykorzystuje najnowszy dodatek do Husk, η(działaj na indeksach). Chodzi o to, że ηpobiera funkcję wyższego rzędu α(tutaj wbudowaną funkcję lambda) i listę xoraz wywołuje αfunkcję indeksowania x(która jest ⁰w powyższym programie) i indeksy x. Na przykład ṁ⁰bierze podzbiór indeksów, mapuje indeksy xnad nimi i sumuje wyniki.

JavaScript (ES6), 96 bajtów

Pobiera dane wejściowe w składni curry (list)(target).

a=>s=>a.reduce((b,_,x)=>[...b,...b.map(y=>[...y,x])],[[]]).filter(b=>!b.reduce((p,i)=>p-a[i],s))

Przypadki testowe

Nie udałoby się to w drugim przypadku testowym, gdyby 4.8 i 10 zostały zamienione z powodu błędu precyzji IEEE 754 - tzn . 14.8 - 4.8 - 10 == 0Ale 14.8 - 10 - 4.8 != 0. Myślę, że to jest w porządku , chociaż może być bardziej odpowiednie odniesienie gdzieś w meta.

let f =

a=>s=>a.reduce((b,_,x)=>[...b,...b.map(y=>[...y,x])],[[]]).filter(b=>!b.reduce((p,i)=>p-a[i],s))

console.log(JSON.stringify(f([1, 2, 1, 5])(8)))
console.log(JSON.stringify(f([4.8, 9.5, 2.7, 11.12, 10])(14.8)))
console.log(JSON.stringify(f([7, 8, 9, -10, 20, 27])(17)))
console.log(JSON.stringify(f([1, 2, 3])(7)))

Skomentował

a => s =>                 // given an array a[] of length N and an integer s
  a.reduce((b, _, x) =>   // step #1: build the powerset of [0, 1, ..., N-1]
    [ ...b,               //   by repeatedly adding to the previous list b[]
      ...b                //   new arrays made of:
      .map(y =>           //     all previous entries stored in y[]
        [...y, x]         //     followed by the new index x
      )                   //   leading to:
    ],                    //   [[]] -> [[],[0]] -> [[],[0],[1],[0,1]] -> ...
    [[]]                  //   we start with a list containing an empty array
  )                       // end of reduce()
  .filter(b =>            // step #2: filter the powerset
    !b.reduce((p, i) =>   //   keeping only entries b
      p - a[i],           //     for which sum(a[i] for i in b)
      s                   //     is equal to s
    )                     //   end of reduce()
  )                       // end of filter()

Python 2 , 110 bajtów

lambda a,n:[b for b in[[i for i in range(len(a))if j&1<<i]for j in range(2**len(a))]if sum(a[c]for c in b)==n]

Wypróbuj online!

R , 85 84 bajtów

function(l,k){N=combn
o={}
for(i in I<-1:sum(l|1))o=c(o,N(I,i,,F)[N(l,i,sum)==k])
o}

Wypróbuj online!

1-indeksowany.

combnzwykle zwraca a matrix, ale ustawienie simplify=Fzwraca listzamiast tego, co pozwala nam na cłączenie wszystkich wyników razem. combn(I,i,,F)zwraca wszystkie kombinacje indeksów i bierzemy N(l,i,sum)==kjako indeks do tej listy, aby określić te, które są równe k.

J , 32 31 bajtów

(=1#.t#])<@I.@#t=.1-[:i.&.#.1"0

Wypróbuj online!

                  1-[:i.&.#.1"0         Make a list of all masks
                                        for the input list. We flip the bits
                                        to turn the unnecessary (0...0)         
                                        into (1...1) that would be missing.
                                        Define it as t.

(=1#.t#])                               Apply the masks, sum and
                                        compare with the target

         <@I.@#                         Turn the matching masks into 
                                        lists of indices

Japt , 14 bajtów

m, à f_x!gU ¥V

Przetestuj online!

Jak to działa

m, à f_x!gU ¥V   Implicit: U = input array, V = target sum
m,               Turn U into a range [0, 1, ..., U.length - 1].
   à             Generate all combinations of this range.
     f_          Filter to only the combinations where
       x           the sum of
        !gU        the items at these indices in U
            ¥V     equals the target sum.
                 Implicit: output result of last expression

Czysty , 104 102 98 bajtów

import StdEnv
@n=[[]:[[i:b]\\i<-[0..n-1],b<- @i]]
?l n=[e\\e<-tl(@(length l))|sum(map((!!)l)e)==n]

Wypróbuj online!

Haskell , 76 bajtów

x#t=[i|i<-tail$concat<$>mapM(\z->[[],[z]])[0..length x-1],t==sum[x!!k|k<-i]]

Wypróbuj online!

Galaretka , 11 bajtów

ị³S=
JŒPçÐf

Wypróbuj online!

1-indeksowany. 4 bajty wydane na zwracanie indeksów, a nie tylko same elementy.

-1 bajt dzięki użytkownikowi202729
-1 bajt dzięki Jonathan Allan

Wolfram Language (Mathematica) , 43 bajty

1-indeksowany.

Pick[s=Subsets;s@Range@Tr[1^#],Tr/@s@#,#2]&

Wypróbuj online!

Python 3 , 144 bajty

lambda a,t:[[e for e,_ in x]for r in range(len(a))for x in combinations(list(enumerate(a)),r+1)if sum(y for _,y in x)==t]
from itertools import*

Wypróbuj online!

0-indeksowane. 44 bajty wydane na zwracanie indeksów, a nie tylko same elementy.

Brachylog , 18 15 bajtów

hiᶠ⊇Shᵐ+~t?∧Stᵐ

Wypróbuj online!

-3 bajty, ponieważ teraz działa jako generator . (Prawdopodobnie można grać w golfa więcej, ale obejście potrzeby używania indeksów jest niewygodne.)

    S              The variable S
   ⊇               is a sublist of
  ᶠ                the list of all
 i                 pairs [element, index] from
h                  the first element of
                   the input;
     hᵐ            the first elements of each pair
       +           add up to
        ~t         the last element of
          ?        the input
           ∧       which isn't necessarily
            S      S,
             tᵐ    from which the last elements of each pair
                   are output.

Perl 6 , 45 bajtów

->\a,\b{grep {a[$_].sum==b},^a .combinations}

Sprawdź to

Rozszerzony:

->
  \a, # input list
  \b, # input target
{

  grep

  {
      a[ $_ ].sum # use the list under test as indexes into ｢a｣
    ==
      b
  },

  ^a              # Range upto ｢a｣ (uses ｢a｣ as a number)
  .combinations   # get all of the combinations
}

APL (NARS), 49 znaków, 98 bajtów

{∨/b←⍺=+/¨{∊⍵⊂w}¨n←{⍵⊤⍨k⍴2}¨⍳¯1+2*k←≢w←⍵:⍸¨b/n⋄⍬}

1-indeksowany; test:

  f←{∨/b←⍺=+/¨{∊⍵⊂w}¨n←{⍵⊤⍨k⍴2}¨⍳¯1+2*k←≢w←⍵:⍸¨b/n⋄⍬}
  ⎕fmt 8 f 1 2 1 5
┌2──────────────┐
│┌3────┐ ┌3────┐│
││2 3 4│ │1 2 4││
│└~────┘ └~────┘2
└∊──────────────┘
  ⎕fmt   14.8  f  4.8 9.5 2.7 11.12 10
┌1────┐
│┌2──┐│
││1 5││
│└~──┘2
└∊────┘
  ⎕fmt 17 f 7, 8, 9, ¯10, 20, 27
┌3──────────────────┐
│┌2──┐ ┌2──┐ ┌3────┐│
││4 6│ │2 3│ │1 4 5││
│└~──┘ └~──┘ └~────┘2
└∊──────────────────┘
  ⎕fmt 7 f 1 2 3
┌0┐
│0│
└~┘

komentarz:

{∨/b←⍺=+/¨{∊⍵⊂w}¨n←{⍵⊤⍨k⍴2}¨⍳¯1+2*k←≢w←⍵:⍸¨b/n⋄⍬}
                             ⍳¯1+2*k←≢w←⍵         copy ⍵ in w, len(⍵) in k, return 1..2^(k-1) 
                 n←{⍵⊤⍨k⍴2}¨                     traslate in binary each element of  1..2^(k-1) and assign the result to n
          {∊⍵⊂w}¨                                for each binary element of n return the elemets of ⍵ in the place where there are the 1s
    b←⍺=+/¨                                       sum them and see if the sum is ⍺, that binary array saved in b
  ∨/                                     :⍸¨b/n   if or/b, get all the elements of n that are 1s for array b, and calculate each all indexs
                                               ⋄⍬ else return Zilde as void set

Pyth, 11 bajtów

fqvzs@LQTyU

Spróbuj go online tutaj , lub sprawdzić wszystkie przypadki testowe od razu tutaj .

fqvzs@LQTyUQ   Implicit: Q=input 1 (list of numbers), z=input 2 (target value, as string)
               Trailing Q inferred
          UQ   Generate range [0-<length of Q>)
         y     Powerset of the above
f              Keep elements of the above, as T, when the following is truthy:
      L T        Map elements of T...
     @ Q         ... to the indicies in Q
    s            Take the sum
 q               Is the above equal to...
  vz             z as an integer
               Implicit print of the remaining results