Генерация комбинаций, которые складываются в целевое значение

14

Вызов

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

Вход и выход

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

Примеры

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: [ ]

счет

Это , поэтому выигрывает самый короткий код!

soapergem
источник
6
Связанный , возможно, обман.
Джузеппе
Я думаю, что это обман, но я бы предпочел закрыть старый, потому что он устарел.
Пост Рок Гарф Хантер
4
Действительно ли числа с плавающей запятой что-то добавляют к задаче? Не уверен, что такое консенсус, но они, вероятно, приведут к ошибкам точности во многих языках.
Арнаулд
Я намеревался учесть плавающие точки, да
soapergem
14
Хм, показатели? Я думаю, что это было бы более приятной задачей, возвращая список значений, хотя я предполагаю, что возникает вопрос о том, как повторяются значения в подмножествах.
xnor

Ответы:

3

Шелуха , 10 байт

ηλfo=¹ṁ⁰tṖ

1-индексироваться. Попробуйте онлайн!

объяснение

ηλ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.

Это использует последнее дополнение к Husk, η(действуют на индексы). Идея состоит в том, что она ηпринимает функцию более высокого порядка α(здесь - встроенную лямбда-функцию) и список x, а также вызывает αфункцию индексации x(которая находится в приведенной выше программе) и индексы x. Например, ṁ⁰берет подмножество индексов, сопоставляет индексирование с xними и суммирует результаты.

Zgarb
источник
9

JavaScript (ES6), 96 байт

Принимает ввод в синтаксисе карри (list)(target).

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

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

Это потерпит неудачу во втором тестовом случае, если 4.8 и 10 были поменяны местами из-за ошибки точности IEEE 754 - то есть, 14.8 - 4.8 - 10 == 0но 14.8 - 10 - 4.8 != 0. Я думаю, что это хорошо , хотя где-то в мета может быть более релевантная ссылка.

комментарии

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()
Arnauld
источник
7
Не один, а два reduceс? Я должен выразить это.
Нил
1
@Neil Меньше известный "reduMapReduce"
Лорд Фаркуаад
7

R , 85 84 байта

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}

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

1-индексироваться.

combnобычно возвращает a matrix, но параметр simplify=Fвозвращает listвместо этого, что позволяет cобъединить все результаты вместе. combn(I,i,,F)возвращает все комбинации индексов, и мы берем N(l,i,sum)==kв качестве индекса этот список, чтобы определить те, которые равны k.

Giuseppe
источник
7

J , 32 31 байт

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

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

                  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
FrownyFrog
источник
Я чувствую , как явное определение помогло бы данному все композиций, но , к сожалению , я только получил ту же длину: 4 :'<@I.t#~x=1#.y#~t=.#:}.i.2^#y'. Попробуйте онлайн!
Коул
5

Japt , 14 байт

m, à f_x!gU ¥V

Проверьте это онлайн!

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

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
ETHproductions
источник
Хороший трюк с m,. Я имел Êo à k@VnXx@gXдля того же количества байтов.
Лохматый
5

Чистый , 104 102 98 байт

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

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

Οurous
источник
[1, 2, -1, 5] 0 --> [[],[2,0]] Требуется набор непустых списков.
FrownyFrog
@FrownyFrog Fixed
Οurous
4

Haskell , 76 байт

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

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

Кристиан Лупаску
источник
[1, 2, -1, 5]#0 --> [[],[0,2]]Требуется набор непустых списков.
FrownyFrog
@FrownyFrog Спасибо! Исправлена.
Кристиан Лупаску
4

Желе , 11 байт

ị³S=
JŒPçÐf

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

1-индексироваться. 4 байта расходуются на возвращение индексов, а не только самих элементов.

-1 байт благодаря пользователю 202729
-1 байт благодаря Джонатану Аллану

HyperNeutrino
источник
12 байтов
user202729
@ user202729 Круто, спасибо!
HyperNeutrino
1
-1 байт: не требуется , если вы используете çвместо Ç.
Джонатан Аллан
@JonathanAllan о, хороший улов спасибо!
HyperNeutrino
2

Python 3 , 144 байта

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*

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

0 индексированные. 44 байта ушло на возвращение индексов, а не только самих элементов.

HyperNeutrino
источник
2

Брахилог , 18 15 байт

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

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

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

    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.
Несвязанная строка
источник
hiᶠ⊇z+ʰXh~t?∧Xtвыходит на такую ​​же длину.
Несвязанная Строка
1

Perl 6 , 45 байт

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

Проверь это

Expanded:

->
  \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
}
Брэд Гилберт b2gills
источник
1

APL (NARS), 49 символов, 98 байтов

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

1-индексированных; тестовое задание:

  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│
└~┘

комментарий:

{∨/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
RosLuP
источник
0

Pyth, 11 байт

fqvzs@LQTyU

Попробуйте онлайн здесь или проверьте все тестовые примеры сразу здесь .

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
Sok
источник