[BOJ]1920個の数字を検索


質問する
N個の整数A[1]、A[2]、...、およびA[N]が与えられた場合、Xという整数が存在するか否かを判定するプログラムを作成してください.
入力
第1行は自然数N(1≦N≦100000)を与える.次の行には、N個の整数A[1],A[2],...,A[N]が与えられる.次の行はM(1≦M≦100000)を与える.次の行はM個の数を与え、これらの数がAに存在するかどうかを見つければよい.すべての整数の範囲は、231未満の-231以上です.
しゅつりょく
M行に答えを出力します.存在する場合は1を出力し、存在しない場合は0を出力します.
入力例1
5
4 1 5 2 3
5
1 3 7 9 5
サンプル出力1
1
1
0
0
1
コード#コード#
古いコード
import java.io.*;
import java.util.Hashtable;

public class Main {
    public static void main(String[] args) throws IOException {
	// TODO Auto-generated method stub
	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        Hashtable<Integer, String> ht = new Hashtable<>();
        String[] A, B;

        int N = Integer.parseInt(br.readLine());
        String s = br.readLine();
        int M = Integer.parseInt(br.readLine());
        String ss = br.readLine();
        br.close();

        A = s.split(" ");
        B = ss.split(" ");
        for (int i = 0; i < N; i++) {
            ht.put(i, A[i]);
        }

        for (int i = 0; i < M; i++) {
            if (ht.containsValue(B[i])) {
                bw.write(1 + "\n");
            } else {
                bw.write(0 + "\n");
            }
        }

        bw.flush();
        bw.close();
    }

}
->タイムアウト
正しいコード
import java.io.*;
import java.util.HashMap;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        HashMap<String, Integer> hm = new HashMap<>();

        int n = Integer.parseInt(br.readLine());
        String ns = br.readLine();
        int m = Integer.parseInt(br.readLine());
        String ms = br.readLine();

        String nArray[] = ns.split(" ");
        String mArray[] = ms.split(" ");

        for (String ss : nArray) {
            hm.put(ss, 0);
        }

        for (String ss : mArray) {
            if (hm.containsKey(ss)) {
                bw.write("1\n");
            } else {
                bw.write("0\n");
            }
        }

        bw.flush();
    }
}
HashTableとHashMap,bwを比較した.write部分、close()部分は処理時間にあまり影響しません.
最も影響が大きいのは、「比較する値がキー値か値か」です.
以前はハッシュも用いていたが,value値が利用されるためタイムアウトが必要になる可能性がある.
単純な情報が含まれているかどうかだけを知っている場合は、valueではなくkeyを使用する必要があります.
にぶんこうほうコード
package baekjoon;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class BOJ1920 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());
        int[] firstArr = new int[n];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            firstArr[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(firstArr);

        int m = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());
        int curNum, answer, left, right, mid;
        for (int i = 0; i < m; i++) {
            curNum = Integer.parseInt(st.nextToken());
            answer = 0;
            left = 0;
            right = n - 1;

            while (left <= right) {
                mid = left + (right - left) / 2;
                if (curNum > firstArr[mid]) {
                    left = mid + 1;
                } else if (curNum < firstArr[mid]) {
                    right = mid - 1;
                } else {
                    answer = 1;
                    break;
                }
            }
            System.out.println(answer);
        }
    }
}