Unificação de matrizes

24

Introdução

Considere duas matrizes do mesmo comprimento, digamos A = [0,1,0,2]e B = [-1,1,2,2]. Suponha que saibamos que seu conteúdo é equivalente em algum sentido, item por item:

  • 0é equivalente a -1,
  • 1é equivalente a 1,
  • 0é equivalente a 2, e
  • 2é equivalente a 2.

A equivalência é transitória: -1e 0são equivalentes, e 0e 2são equivalentes, por isso -1e 2também são equivalentes. A unificação de Ae Bé a matriz em que cada item de A(ou B) foi substituído pelo maior número equivalente a ele. Nesse caso, a unificação seria [2,1,2,2].

A tarefa

Escreva um programa ou função que use duas matrizes inteiras não vazias de igual comprimento e produza sua unificação. Você também pode modificar uma das entradas no lugar em vez de retornar. A menor contagem de bytes vence.

Casos de teste

[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
fonte
3
Não sei bem por que você chamou essa operação de unificação.
Fatalize 11/11
4
@Fatalize Fui inspirado pela unificação de tipos .
Zgarb

Respostas:

6

JavaScript (ES6), 100 90 110 102 96 bytes

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={}))

Minha solução inicial foi de 90 bytes:

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={}))

Embora esteja passando em todos os casos de teste fornecidos, ele falha em algo como:

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

Casos de teste

Mostrar snippet de código

Arnauld
fonte
Isso é um monte de a.map...
ETHproductions
@ETHproductions Yup. Pode haver uma maneira melhor. Fato levemente interessante: os dois primeiros também a.mappodem ser substituídos por b.map.
Arnauld
Eu adicionei um novo caso de teste para sua situação.
Zgarb
5

CJam , 27 bytes

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

Experimente online! Suíte de teste.

Explicação

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.
Martin Ender
fonte
4

Python 2, 91 bytes

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]
Mitch Schwartz
fonte
4

Python, 86 bytes

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])

Atualiza simultaneamente as duas listas, substituindo cada valor na primeira lista pelo elemento correspondente na segunda lista, se for maior. A substituição é feita mapno getmétodo de um dicionário . Em seguida, alterna as listas e repete até que sejam iguais.

xnor
fonte
2

Pitão, 13 bytes

eMumS{s@#dGGC

Experimente online: Demonstração

Explicação:

Comece com cada par. Estenda iterativamente cada par (lista) com listas sobrepostas, desduplique os elementos e classifique. Pare quando este processo convergir. Imprima o máximo de cada lista.

Jakube
fonte
2

PHP, 266 241 213 200 bytes

Solução:

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

Uso: u([1,2,3], [0,0,1]);retorna a matriz desejada.

Não tão jogado:

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
fonte
1

Mathematica, 56 bytes

#/.($|##->Max@##&@@@ConnectedComponents@Thread[#<->#2])&
alefalpha
fonte
0

Java, 273 263 bytes

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

O método f(int[]a,int[]b)resolve o desafio.

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

}

Primeiro, passe pelas duas matrizes e acompanhe os números equivalentes. Em seguida, modifique todos os elementos da primeira matriz para que os números equivalentes sejam armazenados.

Jack Munição
fonte
0

Python, 522 bytes

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)

Explicação

Faça uma tabela de valores correspondentes a cada elemento exclusivo em ambas as matrizes ( ae bneste caso). Por exemplo, se

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

então a tabela seria:

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

em seguida, aplique a pesquisa em profundidade primeiro, então, por exemplo, suponha que eu escolha o elemento mais à esquerda no avalor seja then 0e 0tenha as equivalências: 0e 2. Como 0já foi visitado, vá para 2. 2 tem as equivalências: 0. Portanto, o melhor resultado para escolher o elemento mais à esquerda aé 2. Aqui está a árvore:

   0   
 /   \
0     2
       \
        0

e você deseja levar o maior valor possível para o resultado 2.

Bobas_Pett
fonte
Bem-vindo ao PPCG! No code-golf , você deseja obter o menor número de bytes possível em seu programa. Isso significa encurtar nomes de funções e variáveis ​​e remover espaços em branco desnecessários no seu programa.
Kritixi Lithos
Você também deve usar as duas matrizes como entrada do usuário em vez de codificá-las.
Zgarb
0

PHP, 132 bytes

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];}

Função anônima que recebe duas matrizes.

Esta é a minha opinião sobre 'modificar uma das matrizes no local', conforme especificado na saída do desafio. Isso percorre cada uma das duas matrizes, registra a equivalência se a atual for maior que a armazenada, depois percorre a primeira matriz e substitui todos os valores pelos seus maiores equivalentes. A primeira matriz é tomada por referência (daí a &$a), então a matriz passada é modificada 'no lugar'.

Xanderhall
fonte
0

Java, 170 bytes

Golfe

(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!
}

Função anônima que recebe dois int[]s como argumentos e retorna um int[].

Xanderhall
fonte