Bagaimana cara menghasilkan peta dengan nilai berbeda dari peta (dan menggunakan kunci kanan menggunakan BinaryOperator)?

13

Saya memiliki peta Map<K, V>dan tujuan saya adalah untuk menghapus nilai duplikat dan menampilkan struktur yang sama Map<K, V>lagi. Dalam hal nilai duplikat ditemukan, harus ada satu kunci yang dipilih ( k) dari dua kunci ( k1dan k1) yang memegang nilai-nilai ini, karena alasan ini, asumsikan BinaryOperator<K>pemberian kdari k1dan k2tersedia.

Contoh input dan output:

// Input
Map<Integer, String> map = new HashMap<>();
map.put(1, "apple");
map.put(5, "apple");
map.put(4, "orange");
map.put(3, "apple");
map.put(2, "orange");

// Output: {5=apple, 4=orange} // the key is the largest possible

Usaha saya menggunakan Stream::collect(Supplier, BiConsumer, BiConsumer)adalah sedikit sangat kikuk dan berisi operasi bisa berubah seperti Map::putdan Map::removeyang saya ingin menghindari:

// // the key is the largest integer possible (following the example above)
final BinaryOperator<K> reducingKeysBinaryOperator = (k1, k2) -> k1 > k2 ? k1 : k2;

Map<K, V> distinctValuesMap = map.entrySet().stream().collect(
    HashMap::new,                                                              // A new map to return (supplier)
    (map, entry) -> {                                                          // Accumulator
        final K key = entry.getKey();
        final V value = entry.getValue();
        final Entry<K, V> editedEntry = Optional.of(map)                       // New edited Value
            .filter(HashMap::isEmpty)
            .map(m -> new SimpleEntry<>(key, value))                           // If a first entry, use it
            .orElseGet(() -> map.entrySet()                                    // otherwise check for a duplicate
                    .stream() 
                    .filter(e -> value.equals(e.getValue()))
                    .findFirst()
                    .map(e -> new SimpleEntry<>(                               // .. if found, replace
                            reducingKeysBinaryOperator.apply(e.getKey(), key), 
                            map.remove(e.getKey())))
                    .orElse(new SimpleEntry<>(key, value)));                   // .. or else leave
        map.put(editedEntry.getKey(), editedEntry.getValue());                 // put it to the map
    },
    (m1, m2) -> {}                                                             // Combiner
);

Apakah ada solusi menggunakan kombinasi yang tepat Collectorsdalam satu Stream::collectpanggilan (mis. Tanpa operasi yang bisa berubah)?

Nikolas
sumber
2
Apa metrik Anda untuk " lebih baik " atau " terbaik "? Haruskah dilakukan melalui Streams?
Turing85
Jika nilai yang sama dikaitkan dengan 2 kunci, bagaimana Anda memilih kunci mana yang dipertahankan?
Michael
Apa hasil yang diharapkan dalam kasus Anda?
YCF_L
1
@ Turing85: Seperti yang saya katakan. Yang lebih baik atau yang terbaik adalah tanpa penggunaan eksplisit dari metode peta yang bisa berubah seperti Map::putatau Map::removedi dalam Collector.
Nikolas
1
Perlu dicoba BiMap. Kemungkinan merupakan duplikat Hapus nilai duplikat dari HashMap di Java
Naman

Jawaban:

12

Anda dapat menggunakan Collectors.toMap

private Map<Integer, String> deduplicateValues(Map<Integer, String> map) {
    Map<String, Integer> inverse = map.entrySet().stream().collect(toMap(
            Map.Entry::getValue,
            Map.Entry::getKey,
            Math::max) // take the highest key on duplicate values
    );

    return inverse.entrySet().stream().collect(toMap(Map.Entry::getValue, Map.Entry::getKey));
}
MikeFHay
sumber
9

Coba ini: Cara sederhana adalah membalikkan kunci dan nilai kemudian gunakan toMap()kolektor dengan fungsi gabungan.

map.entrySet().stream()
        .map(entry -> new AbstractMap.SimpleEntry<>(entry.getValue(), entry.getKey()))
        .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, reducingKeysBinaryOperator));

Map<K, V> output = map.entrySet().stream()
        .collect(Collectors.toMap(Map.Entry::getValue, Map.Entry::getKey, reducingKeysBinaryOperator))
        .entrySet().stream()
        .collect(Collectors.toMap(Map.Entry::getValue, Map.Entry::getKey));
Hadi J
sumber
2
Saya gagal melihat apa yang mapdibeli operasi perantara . Anda tampaknya menukar kunci dan nilai, itu jelas, tapi apa gunanya, Anda bisa melakukan itu pada langkah kumpulkan sama saja?
GPI
3
@ GPI dan Michael, ini karena dia harus menggabungkan kunci, sehingga membalikkan pasangan akan menggabungkan kunci. Apa yang hilang adalah inversi kedua.
Jean-Baptiste Yunès
2
@HadiJ Tidak! Pembalikan benar! tetapi yang kedua diperlukan untuk kembali. Merge digunakan untuk menggabungkan kunci, tetapi penggabungan hanya mungkin untuk nilai ...
Jean-Baptiste Yunès
@ Jean-BaptisteYunès Saya mengerti kebutuhan untuk menggabungkan, tetapi mengapa saya tidak segera mendapatkan adalah mengapa Anda kode swap(); collect(key, value, binOp);alih-alih collect(value, key, binOp). Mungkin saya perlu mencoba ini secara singkat?
GPI
2
Mengambil kebebasan untuk menggunakan variabel lokal yang diperkenalkan dalam pertanyaan dalam kode yang dibagikan oleh Anda. Kembalikan kalau-kalau itu bertentangan dengan niat saat Anda membuat jawabannya.
Naman
4

Saya menemukan solusi non-stream lebih ekspresif:

BinaryOperator<K> reducingKeysBinaryOperator = (k1, k2) -> k1 > k2 ? k1 : k2;

Map<V, K> reverse = new LinkedHashMap<>(map.size());
map.forEach((k, v) -> reverse.merge(v, k, reducingKeysBinaryOperator));

Map<K, V> result = new LinkedHashMap<>(reverse.size());
reverse.forEach((v, k) -> result.put(k, v));

Ini digunakan Map.mergedengan mengurangi dua fungsi dan digunakan LinkedHashMapuntuk mempertahankan urutan entri asli.

Federico Peralta Schaffner
sumber
2
Ya, saya telah menyimpulkan solusi (serupa) ini. Namun, saya mencari pendekatan java-stream , karena ini adalah cara yang lebih deklaratif. Dapatkan +1 saya
Nikolas
1

Saya menemukan cara menggunakan hanya Collectorstanpa perlu mengumpulkan dan memproses lebih lanjut Peta yang dikembalikan lagi. Idenya adalah:

  1. Kelompokkan Map<K, V>ke Map<V, List<K>.

    Map<K, V> distinctValuesMap = this.stream.collect(
        Collectors.collectingAndThen(
            Collectors.groupingBy(Entry::getValue),
            groupingDownstream 
        )
    );

    {apple = [1, 5, 3], oranye = [4, 2]}

  2. Kurangi kunci baru ( List<K>) untuk Kmenggunakan BinaryOperator<K>.

    Function<Entry<V, List<Entry<K, V>>>, K> keyMapFunction = e -> e.getValue().stream()
        .map(Entry::getKey)
        .collect(Collectors.collectingAndThen(
            Collectors.reducing(reducingKeysBinaryOperator),
            Optional::get
        )
    );

    {apple = 5, orange = 4}

  3. Balikkan Map<V, K>kembali ke Map<K, V>struktur lagi - yang aman karena kedua kunci dan nilai dijamin berbeda.

    Function<Map<V, List<Entry<K,V>>>, Map<K, V>> groupingDownstream = m -> m.entrySet()
        .stream()
        .collect(Collectors.toMap(
            keyMapFunction,
            Entry::getKey
        )
    );

    {5 = apel, 4 = oranye}

Kode terakhir:

final BinaryOperator<K> reducingKeysBinaryOperator = ...

final Map<K, V> distinctValuesMap = map.entrySet().stream().collect(
        Collectors.collectingAndThen(
            Collectors.groupingBy(Entry::getValue),
            m -> m.entrySet().stream().collect(
                Collectors.toMap(
                    e -> e.getValue().stream().map(Entry::getKey).collect(
                        Collectors.collectingAndThen(
                            Collectors.reducing(reducingKeysBinaryOperator),
                            Optional::get
                        )
                    ),
                    Entry::getKey
                )
            )
        )
    );
Nikolas
sumber
1

Pendekatan lain untuk mendapatkan hasil yang diinginkan dengan "Stream and Collectors.groupingBy".

    map = map.entrySet().stream()
    .collect(Collectors.groupingBy(
            Entry::getValue,
            Collectors.maxBy(Comparator.comparing(Entry::getKey))
            )
    )
    .entrySet().stream()
    .collect(Collectors.toMap(
            k -> {
                return k.getValue().get().getKey();
            }, 
            Entry::getKey));
vishesh chandra
sumber