Объединение массивов

24

Введение

Рассмотрим два массива одинаковой длины, скажем, A = [0,1,0,2]и B = [-1,1,2,2]. Предположим, мы знаем, что их содержание в некотором смысле эквивалентно, элемент за элементом:

  • 0эквивалентно -1,
  • 1эквивалентно 1,
  • 0эквивалентно 2, и
  • 2эквивалентно 2.

Эквивалентность является переходным: -1и 0эквивалентны, и 0и 2эквивалентны, так -1и 2также эквивалентны. Объединение из Aи Bявляется массив , в котором каждый элемент A(или B) был заменен наибольшим числом , что это эквивалентно. В этом случае объединение будет [2,1,2,2].

Задание

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

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

[0] [0] -> [0]
[1] [2] -> [2]
[0,-1] [-1,-1] -> [0,0]
[0,1,0] [2,1,0] -> [2,1,2]
[1,2,3] [0,0,1] -> [3,3,3]
[0,1,0,2] [-1,1,2,2] -> [2,1,2,2]
[1,0,1,-4] [-3,-1,-2,2] -> [1,0,1,2]
[1,2,3,-2] [1,0,-3,-2] -> [1,2,3,-2]
[-3,-2,-1,0,1] [-1,-1,-1,-1,-1] -> [1,1,1,1,1]
[-3,-2,-1,0,1] [2,-1,0,1,-3] -> [2,2,2,2,2]
[-3,5,5,3,1] [4,2,3,1,2] -> [4,5,5,5,5]
[4,0,2,-5,0] [0,4,-5,3,5] -> [5,5,3,3,5]
[-2,4,-2,3,2,4,1,1] [-2,4,1,2,2,3,1,-2] -> [1,4,1,4,4,4,1,1]
[-10,-20,-11,12,-18,14,-8,-1,-14,15,-17,18,18,-6,3,1,15,-15,-19,-19] [-13,6,-4,3,19,1,-10,-15,-15,11,6,9,-11,18,6,6,-5,-15,7,-11] -> [-8,14,18,14,19,14,-8,-1,-1,15,14,18,18,18,14,14,15,-1,18,18]
[20,15,2,4,-10,-4,-19,15,-5,2,13,-3,-18,-5,-6,0,3,-6,3,-17] [-18,7,6,19,-8,-4,-16,-1,13,-18,8,8,-16,17,-9,14,-2,-12,7,6] -> [20,15,20,19,-8,-4,20,15,17,20,17,17,20,17,-6,14,15,-6,15,20]
code-golf array-manipulation Zgarb
источник
3
Я не совсем уверен, почему вы назвали эту операцию объединением.
Fatalize
4
@Fatalize Я был вдохновлен объединением типов .
Згарб

Ответы:

6

JavaScript (ES6), 100 90 110 102 96 байт

a=>b=>a.map(v=>t[v],a.map((_,k)=>a.map((x,i)=>t[x]=t[y=b[i]]=Math.max(k?t[x]:x,k?t[y]:y)),t={}))

Мое первоначальное решение было 90 байтов:

a=>b=>a.map(v=>t[v],a.map(_=>a.map((x,i)=>t[x]=t[y=b[i]]=Math.max(t[x]||x,t[y]||y)),t={}))

Несмотря на то, что он проходит все предоставленные тестовые случаи, он не может выполнить что-то вроде:

A = [0, -1], B = [-1, -1]

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

Показать фрагмент кода

Arnauld
источник
Это много a.map...
ETHproductions
@ETHproductions Да. Там может быть лучший способ. Мягко интересный факт: первые два a.mapможно заменить на b.mapточно такие же.
Арнаулд
Я добавил новый тестовый пример для вашей ситуации.
Згарб
5

CJam , 27 байт

l~_])\z_,*f{{_2$&,*|}/:e>}p

Попробуйте онлайн! Тестирование.

объяснение

l~       e# Read and evaluate input, dumping arrays A and B on the stack.
_        e# Copy B.
])\      e# Wrap in array, pull off B, swap. Gives B [A B] on the stack.
z        e# Transpose the [A B] matrix to get a list of all equivalent pairs.
_,*      e# Repeat this list by the number of pairs. This is to ensure that the
         e# following procedure is applied often enough to allow transitive
         e# equivalences to propagate.
f{       e# Map this block over B, passing in the list of pairs each time...
  {      e#   For each pair...
    _2$  e#     Copy both the pair and the current value/list.
    &,   e#     Get the length of their intersection. If this is non-zero,
         e#     the current pair belongs to the current equivalence class.
    *    e#     Repeat the pair that many times.
    |    e#     Set union between the current value/list and the repeated pair.
         e#     This adds the pair to the current list iff that list already
         e#     contains one value from the pair.
  }/
  :e>    e#   Get the maximum value of this equivalence class.
}
p        e# Pretty print.
Мартин Эндер
источник
4

Python 2, 91 байт

f=lambda a,b:[a<x>b.update(b&set(x)and x)and b or max(f(zip(a,b)*len(a),{x})[0])for x in a]
Митч Шварц
источник
4

Python, 86 байт

f=lambda a,b:a*(a==b)or f(*[map({x:y for x,y in zip(a,b)if x<y}.get,x,x)for x in b,a])

Одновременно обновляет оба списка, заменяя каждое значение в первом списке соответствующим элементом во втором списке, если оно больше. Замена выполняется mapпо getметоду словаря . Затем меняет списки и повторяет до тех пор, пока они не станут равными.

XNOR
источник
2

Pyth, 13 байт

eMumS{s@#dGGC

Попробуйте онлайн: демонстрация

Объяснение:

Начните с каждой пары. Итеративно расширяйте каждую пару (список) перекрывающимися списками, дедуплицируйте элементы и сортируйте. Остановитесь, как только этот процесс сходится. Напечатайте максимум каждого списка.

Jakube
источник
2

Php 266 241 213 200 байт

Решение:

function u($x,$y){foreach($x as$i=>$j){$k[$y[$i]][]=$j;$k[$j][]=$y[$i];}$h=function($c,&$w)use($k,&$h){$w[]=$c;foreach($k[$c]as$u)!in_array($u,$w)&&$h($u,$w);return max($w);};return array_map($h,$x);}

Использование: u([1,2,3], [0,0,1]);возвращает нужный массив.

Не так играли в гольф

function unify($x, $y)
{
    foreach($x as $i=>$j) {
        $k[$y[$i]][] = $j;
        $k[$j][] = $y[$i];
    }

    $h = function ($c, &$w=[]) use ($k, &$h) {
        $w[] = $c;
        foreach($k[$c] as $u)
            !in_array($u, $w) && $h($u, $w);
        return max($w);
    };

    return array_map($h, $x);
}
Progrock
источник
1

Mathematica, 56 байт

#/.($|##->Max@##&@@@ConnectedComponents@Thread[#<->#2])&
alephalpha
источник
0

Ява, 273 263 байта

interface Z{int z(int x);default Z g(int m,int n){return x->{for(int t;x!=(t=x==m?z(n):z(x));)x=t;return x;};}static void f(int[]a,int[]b){Z y=x->x;int i=0,v;for(int u:a){u=y.z(u);v=y.z(b[i++]);if(u<v)y=y.g(u,v);if(v<u)y=y.g(v,u);}i=0;for(int u:a)a[i++]=y.z(u);}}

Метод f(int[]a,int[]b)решает проблему.

interface Z{

  //should return an "equivalent" integer
  int z(int x);

  //return a Z lambda where the 1st arg is "equivalent" to the 2nd arg
  default Z g(int m,int n){
    return x->{
      for(int t;x!=(t=x==m?z(n):z(x));) //always find the last equivalent number for x
        x=t;
      return x;
    };
  }

  //solve the challenge
  static void f(int[]a,int[]b){
    Z y=x->x; //start off with all numbers only equivalent only to themselves
    int i=0,v;
    for(int u:a){
      u=y.z(u); //get a's element's equivalent number
      v=y.z(b[i++]); //get b's element's equivalent number
      if(u<v)y=y.g(u,v); //if a's was smaller than b's, make a's equivalent to b's
      if(v<u)y=y.g(v,u); //if b's was smaller than a's, make b's equivalent to a's
    }
    i=0;
    for(int u:a) //overwrite a with each element's equivalent value
      a[i++]=y.z(u);
  }

}

Сначала пройдитесь по обоим массивам и отследите эквивалентные числа. Затем измените каждый элемент в первом массиве, чтобы сохранить эквивалентные числа.

Джек Боеприпасы
источник
0

Python, 522 байта

a = [-2,4,-2,3,2,4,1,1]
b = [-2,4,1,2,2,3,1,-2]
m = {}
visited = {}
for i in range(len(a)):
    if a[i] in m:
        if b[i] not in m[a[i]]:
            m[a[i]].append(b[i])
    else:
        l = []
        l.append(b[i])
        m[a[i]] = l
    if b[i] in m:
        if a[i] not in m[b[i]]:
            m[b[i]].append(a[i])
    else:
        l = []
        l.append(a[i])
        m[b[i]] = l

def dfs(v, maximum):
    if v > maximum:
        maximum = v
    visited[v] = True
    for n in m[v]:
        if not visited[n]:
            d = dfs(n, maximum)
            if d > v:
                maximum = d
    return maximum

result = []
for k in range(len(a)):
    for q in m:
        visited[q] = False
    result.append(max(dfs(a[k], a[k]), dfs(b[k], b[k])))

print(result)

объяснение

Составьте таблицу значений, соответствующих каждому уникальному элементу в обоих массивах ( aи bв этом случае). Например, если

a = [0,1,0] 
b = [2,1,0]

тогда таблица будет:

0:[0,2]
1:[1]
2:[0]

затем примените поиск в глубину, например, предположим, что я выбрал самый левый элемент в aзначении, то есть 0и 0имеет эквивалентность: 0и 2. Так 0как уже был посещен, перейдите к 2. 2 имеют эквивалентности: 0. Таким образом , лучший результат для выбора самого левого элемента в aIS 2. Вот дерево:

   0   
 /   \
0     2
       \
        0

и вы хотите получить наибольшее значение там, так что результат 2.

Bobas_Pett
источник
Добро пожаловать в PPCG! В код-гольфе вы стремитесь получить как можно меньший счет в вашей программе. Это означает сокращение имен функций и переменных и удаление ненужных пробелов в вашей программе.
Kritixi Lithos
Вы должны также принять эти два массива как пользовательский ввод вместо жесткого их кодирования.
Згарб
0

PHP, 132 байта

function(&$a,$b){for(;$i<count($a);$i++){$r=$a[$i];$s=$b[$i];$r<$c[$s]?:$c[$s]=$r;$s<$c[$r]?:$c[$r]=$s;}foreach($a as&$v)$v=$c[$v];}

Анонимная функция, которая принимает два массива.

Это мой взгляд на «модифицировать один из массивов на месте», как указано в выходных данных задачи. Это проходит по каждому из двух массивов, записывает эквивалентность, если текущий больше, чем сохраненный, затем проходит по первому массиву и заменяет все значения их наибольшими эквивалентами. Первый массив берется по ссылке (отсюда и &$a), поэтому переданный массив изменяется «на месте».

Xanderhall
источник
0

Java, 170 байт

Golfed

(a,b)->{int[]d=a.clone();for(int i=0,y;i<d.length;i++){y=0;for(int j=0;j<a.length;j++)if(a[j]==d[i]||b[j]==d[i])y=Integer.max(y,Integer.max(a[j],b[j]));d[i]=y;}return d;}

Ungolfed

(a, b) -> {                                        // Two argument lambda
    int[] d = a.clone();                           // We clone our first array for modification
    for (int i = 0,y; i < d.length; i++) {         // Going through the elements of d...
        y = 0;                                     // We initialize our 'highest' equivalent
        for (int j = 0; j < a.length; j++) {       // Going through each of our arrays (a and b)...
            if (a[j] == d[i] || b[j] == d[i]) {    // If either of them is the number we're trying to match for equivalence...
                y = Integer.max(y, Integer.max(a[j], b[j])); // We see if the new number is bigger than the largest we've found.
            }
        }
        d[i] = y;                                  // We then assign the largest equivalent number for the current position in our output array.
    }
    return d; // And return!
}

Анонимная функция, которая принимает два int[]s в качестве аргумента и возвращает int[].

Xanderhall
источник