在java中使用用户定义的数据类型对数组进行排序

vwkv1x7d  于 2021-08-20  发布在  Java
关注(0)|答案(1)|浏览(233)

这个问题来自Geeksforgeks:
以下是链接:https://practice.geeksforgeeks.org/problems/find-all-pairs-whose-sum-is-x5808/1#
q) 给定两个不同元素的大小为n的未排序数组a和大小为m的未排序数组b,任务是从两个数组中查找其和等于x的所有对。

import java.util.*;
import java.lang.*;
import java.io.*;

class pair  {  
    long first, second;  
    public pair(long first, long second)  
    {  
        this.first = first;  
        this.second = second;  
    }  
}

class GFG {
    public static void main(String[] args) throws IOException
    {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int t = Integer.parseInt(br.readLine().trim());
        while(t-->0)
        {
            StringTokenizer stt = new StringTokenizer(br.readLine());

            long N = Long.parseLong(stt.nextToken());
            long M = Long.parseLong(stt.nextToken());
            long X = Long.parseLong(stt.nextToken());
            long A[] = new long[(int)(N)];
            long B[] = new long[(int)(M)];

            String inputLine[] = br.readLine().trim().split(" ");
            for (int i = 0; i < N; i++) {
                A[i] = Long.parseLong(inputLine[i]);
            }
            String inputLine1[] = br.readLine().trim().split(" ");
            for (int i = 0; i < M; i++) {
                B[i] = Long.parseLong(inputLine1[i]);
            }

            Solution obj = new Solution();
            pair [] answer = obj.allPairs(A, B, N, M, X);
            int sz = answer.length;

            if(sz==0)
            System.out.println(-1);
            else{
                StringBuilder output = new StringBuilder();
                for(int i=0;i<sz;i++){
                    if(i<sz-1)
                    output.append(answer[i].first +" "+ answer[i].second + ", ");
                    else
                    output.append(answer[i].first +" "+ answer[i].second);

                }
                System.out.println(output);
            }

        }
    }
}
// } Driver Code Ends

//User function Template for Java

/*
class pair  {  
    long first, second;  
    public pair(long first, long second)  
    {  
        this.first = first;  
        this.second = second;  
    }  
}

* /

class Solution {
    public pair[] allPairs( long A[], long B[], long N, long M, long X) {

        //MY CODE STARTS FROM HERE
        ArrayList<pair> list = new ArrayList<>();
        HashSet<Long> set = new HashSet<>();
        for(long i : A){
            set.add(i);
        }

        for(int i=0;i<M;i++){
            if(set.contains(X-B[i])){
                list.add(new pair(X-B[i], B[i]));
            }
        }

        pair arr[] = new pair[list.size()];
        arr = list.toArray(arr);
        for(int j=0;j<(arr.length-1);j++){
            if(arr[j]>arr[j+1]){
                pair temp = arr[j];
                arr[j] = arr[j+1];
                arr[j+1] = temp;
                j -= 1;
            }

        }
        return arr;
    }
}

我必须按升序对结果进行排序,但是这里数组的数据类型是用户定义的(即pair)。所以我得到了这个错误:
java程序:100:错误:如果(arr[j]>arr[j+1]),则二进制运算符'>'的操作数类型错误{^1第一种类型:对第二种类型:对1错误
请帮我对数组进行排序。注:时间复杂度应为o(nlogn)

l7wslrjt

l7wslrjt1#

既然可以在java列表中使用内置排序,为什么还要使用自定义排序呢。
只用 list.sort(Comparator.comparing(p -> ((pair) p).first).thenComparing(p -> ((pair) p).second)); 对配对列表进行排序。

public pair[] allPairs( long A[], long B[], long N, long M, long X) {
        // Your code goes here
        //MY CODE STARTS FROM HERE
        ArrayList<pair> list = new ArrayList<>();
        HashSet<Long> set = new HashSet<>();
        for(long i : A){
            set.add(i);
        }

        for(int i=0;i<M;i++){
            if(set.contains(X-B[i])){
                list.add(new pair(X-B[i], B[i]));
            }
        }

        list.sort(
                Comparator
                        .comparing(p -> ((pair) p).first)
                        .thenComparing(p -> ((pair) p).second)
        );
        return list.toArray(pair[]::new);
    }

相关问题