不知道如何返回数组

nfeuvbwi  于 2021-08-25  发布在  Java
关注(0)|答案(2)|浏览(374)

找到具有给定和的所有对
给定两个不同元素的大小为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<Long> 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(X-B[i]);
                list.add(B[i]);
            }
        }
        long arr[] = new long[list.size()];
        for(int i=0;i<arr.length;i++){
            arr[i] = list.get(i);
        }
        return arr;
    }
}

我得到这个错误:

prog.java:101: error: incompatible types: long[] cannot be converted to pair[]
        return arr;
               ^
1 error

我如何才能正确地返回答案?

xzv2uavs

xzv2uavs1#

在回答之前,在编写代码时使用描述性变量名是非常重要的。它们可能对你有意义,但像我这样的人,他们不是你,甚至可能是未来的你,很难理解a、b、m、n等的含义。
出现此错误的原因是您的函数 allPairs() 具有pair[]的返回类型,但您正在尝试返回long[]。由于变量名的原因,很难理解您要做什么,但是当您找到所需的结果时,似乎是将第一个和第二个结果分别添加到一个long数组中,这也破坏了pair类的点。下面是一个关于它应该是什么样子的想法:

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);
        return list.toArray();
    }
}
pieyvz9o

pieyvz9o2#

在define方法中,返回类型为pairs,但实际上返回一个长[],它必须是编译错误
您应该按如下方式修改解决方案

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

    //MY CODE STARTS FROM HERE
    ArrayList<Long> 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(X-B[i]);
            list.add(B[i]);
        }
    }
    pair arr[] = new pair[list.size()];
    for(int i=0;i<arr.length;i++){
        arr[i]=new pair(X-B[i], B[i]);
    }
    return arr;
}

}

相关问题