글 읽기 - 차집합 코드에 어떤 문제가 있을까요?

dlwodnsdl 7년 전 0

a와 b집합에 대해 각각 정렬해주고, 각각의 원소를 a[0], b[0]부터 a[i]<b[j] 면 a[i]를 출력하고 i에 1을 더해주고, a[i]>b[j]이면 j에 1을 더해주고 같으면 i,j에 모두 1을 더해주며 순차적으로 비교했습니다. 여러가지 테스트케이스를 돌려봤는데 어떤 케이스에서 오류가 나는지 잘 모르겠네요. 어떻게 수정해야할까요?

import java.io.*;
import java.util.*;
class Main{
    public static void main(String[] args)throws Exception{
    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    	BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    	StringTokenizer s=new StringTokenizer(br.readLine().trim());
    	int n=Integer.parseInt(s.nextToken());
    	int m=Integer.parseInt(s.nextToken());
    	int[] a=new int[n];
    	int[] b=new int[m];
    	s=new StringTokenizer(br.readLine().trim());
    	for(int i=0;i<n;i++){
    		a[i]=Integer.parseInt(s.nextToken());
    	}
    	s=new StringTokenizer(br.readLine().trim());
    	for(int i=0;i<m;i++){
    		b[i]=Integer.parseInt(s.nextToken());
    	}
    	Arrays.sort(a);
    	Arrays.sort(b);
    	int count=0;
    	for(int i=0,j=0;;){
    		if(a[i]<b[j]){
    			bw.write(a[i]+" ");
    			i++;
    			count++;
    			if(i==n){
    				break;
    			}
    		}else if(a[i]>b[j]){
    			j++;
    			if(j==m){
    				for(int k=i;k<n;k++){
    					bw.write(a[k]+" ");
    					count++;
    				}
    				break;
    			}
    		}else{
    			i++;
    			j++;
    			if(i==n){
    				break;
    			}
    			if(j==m){
    				for(int k=i;k<n;k++){
    					bw.write(a[k]+" ");
    					count++;
    				}
    				break;
    			}
    		}
    	}
    	System.out.println(count);
    	bw.close();
    }
}

dlwodnsdl 7년 전 0

system.out.print 먼저하고 buffered writer로 입력했더니 해결되네요...

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

차집합 코드에 어떤 문제가 있을까요?

dlwodnsdl 7년 전 0

dlwodnsdl 7년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말