[JAVA][伯俊17143]釣り王

//메인 메서드 내의 틀: 
int answer = 0; //상어의 사이즈를 더할 변수
validateLocation();//입력을 받은 직후 상어를 물에 풀어준다
int t = 0; //t는 0부터 시작해서 열(time)과 같아질 때 까지 낚시를 한다
int time = C;//낚시꾼은 열을 이동하며 낚시를 하며, 열의 끝에오면 낚시가끝난다-> time== Column

while(t<time){
	t++;//time == 낚시꾼의 위치
	answer+=caught(t); //1. 낚시왕이 오른쪽으로 한칸 이동하며 낚시한다
	swim();		   //2. 상어의 예상 위치 계산
	validateLocation();//3. 상어 맵에 배치
}

/**낚시꾼이 t초에 잡은 상어*/
private static int caught(int t) {
	int caught = 0;
	for(int i = 1; i <=R; i++) {//C는 고정--as t
		if(arr[i][t]!=0) {//가장 가까운 상어 발견 -> arr에는 상어의 index가 저장되어있다
			caught= sharks.get(arr[i][t]).size; //상어의 크기 리턴
			sharks.remove(arr[i][t]); //상어 리스트에서 상어는 빼준다
			arr[i][t] = 0;
			return caught;
		}
		}
		return caught;//해당 열에 상어가 없다면 0 리턴
	}

すべてのソース

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.StringTokenizer;
public class 백준_17143_낚시왕  {
	public static class Shark{
		int r;
		int c;
		int speed;
		int direction;//1상2하3우4좌
		int size;
		//초기 initializing
		public Shark(int r, int c, int speed, int direction, int size) {
			this.r =r;
			this.c = c;
			this.speed=  speed;
			this.size = size;
			this.direction = direction;
		}	
	}
	
	static ArrayList<Shark> sharks;//상어의 정보를 저장할 배열
	static int[][] dd= {{0,0},{-1,0},{1,0},{0,1},{0,-1}};//상하우좌
	static int R,C,time,arr[][];
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		R = Integer.parseInt(st.nextToken());
		C = time = Integer.parseInt(st.nextToken());//time: 총 열만큼 움직이고 마지막 열에서 끝나기 때문에 time== C
		int M  = Integer.parseInt(st.nextToken());//상어의 수
		if(M==0) {System.out.println("0");return;}
		arr = new int[R+1][C+1];
		sharks = new ArrayList<>();
		sharks.add(new Shark(0,0,0,0,0));//dummy data
		for(int m = 0; m<M ; m++) {//상어의 수만큼 정보 들어온다
			//at 0초:
			st = new StringTokenizer(br.readLine());
			int r = Integer.parseInt(st.nextToken());//상어 r 좌표
			int c = Integer.parseInt(st.nextToken());//상어 c 좌표
			int s = Integer.parseInt(st.nextToken());//speed
			int d = Integer.parseInt(st.nextToken());//direction
			int z = Integer.parseInt(st.nextToken());//size
			sharks.add(new Shark(r,c,s,d,z));
		}//end of input
		int answer = 0;
		int t =0;
		validateLocation();//initiate
//		MapInfo();
		while(t<time) {
			t++;
//			System.out.println("낚시 start");
			answer += caught(t);//낚시꾼이 먼저 잡고
//			SharkLocationInfo();
//			MapInfo();
//			System.out.printf("after %d초: ",t);
//			System.out.println("상어가 수영했다");
			swim();//상어 이동위치 잡고
//			SharkLocationInfo();
//			System.out.println("validate 후 맵에 추가");
			validateLocation();//상어 옮기고
//			SharkLocationInfo();
//			MapInfo();
		}
		System.out.println(answer);
	}//end of main
    
	/**3. 낚시꾼이 t초에 잡은 상어*/
	private static int caught(int t) {
		int caught = 0;
		for(int i = 1; i <=R; i++) {//C는 고정--as t
			if(arr[i][t]!=0) {//상어가 있다면 그 상어의 
				//arr[][]에는 상어의 인덱스가 담겨져 있다
				caught= sharks.get(arr[i][t]).size;
				sharks.remove(arr[i][t]);
				arr[i][t] = 0;
				return caught;
			}
		}
		return caught;
	}
    
	/** 2. 상어를 맵에 배치 :1번 메소드에서 업데이트된 상어 위치를 기반으로, 
    	해당 위치에 상어 두마리 이상인지 확인하고 크기가 큰 상어만 배치한다.*/
	private static void validateLocation() {
		//배열 초기화:
		for(int i= 0 ;i<=R; i++) {
			for(int j =0; j<=C ;j++) {
				arr[i][j] = 0;
			}
		}
		int index = sharks.size();
		for(int i =1; i<index; i++) {
			int r = sharks.get(i).r;//상어의 r 위치
			int c = sharks.get(i).c;//상어의 c 위치
//			System.out.printf("%d , %d 상어 생성\n",r,c);//확인용
			if(arr[r][c]==0) {//상어가 없다면 그냥 넣어도 된다
				arr[r][c] = i;//사이즈가 아니라 현재 상어의 인덱스를 넣는다! 사이즈를 넣으면  후에 더 큰 상어가 오면 문제생김
			}
			else if(arr[r][c]!=0) {
//				System.out.println("현재 위치에 상어가 이미 있다");
				if(sharks.get(arr[r][c]).size>sharks.get(i).size) {
//					System.out.println(i+"번째 상어는 퇴출한다 -> 기존애 사이즈가 더 커서 ");
					//sharks.remove(i);//작은 상어는 퇴출..하지말고 0,0으로 보내야 문제 발생 x
					sharks.get(i).c = 0;//먹힌 상어들의 좌표를 0,0으로 지정
					sharks.get(i).r = 0;//먹힌 상어들의 좌표를 0,0으로 지정
				}
				else {//현재 상어가 더 크다!
					int eaten = arr[r][c];//잡아먹힌 상어
					sharks.get(eaten).c = 0;//0,0으로 지정
					sharks.get(eaten).r = 0;//0,0으로 지정
					arr[r][c] = i;//큰 상어가 살아 남았다
				}
			}
		}
	//배열 채워짐	
	}
    
	/**1번 메서드: 상어의 예상 이동 위치를 상어 배열에 업데이트 시킨다.*/
	private static void swim() {
outer:	for(int i = 0; i< sharks.size() ;i ++) {
//원래 좌표: 		System.out.printf("Previous Location: %d,%d\n",sharks.get(i).r,sharks.get(i).c);
			int move = sharks.get(i).speed;//이만큼 움직여야 한다
	
			int nr = sharks.get(i).r;//새 위치 //row: y좌표
			int nc = sharks.get(i).c;//새 위치//col: x좌표
			if(nr == 0 || nc==0) continue outer; //먹힌 상어들은 무시하고 다음 i로 이동
			while (move>0) {
				nr += dd[sharks.get(i).direction][0];//dd[방향][r표시]
				nc += dd[sharks.get(i).direction][1];//dd[방향][c표시]
				//움직이고 나서:
				if(nr<1||nc<1||nr>=R+1||nc>=C+1) {//범위를 넘겼을 경우:
					if(sharks.get(i).direction%2 ==0){sharks.get(i).direction-=1;}//짝수일 경우 -1
					else{sharks.get(i).direction+=1;}//홀수일 경우 +1
					
					nr += dd[sharks.get(i).direction][0]*2;//이미 움직여서는 안되는 곳까지 움직였기 때문에 반대 방향으로 두칸 이동
					nc += dd[sharks.get(i).direction][1]*2;		
				}
				move--;
			}
			//위 while 끝나면 nx 와 ny 를 다시 상어 위치로 업데이트
			sharks.get(i).r = nr;
			sharks.get(i).c = nc;
//새 좌표 확인:		System.out.printf("new location: %d, %d\n",nr,nc);
		}
	}


//Utilities
/**상어 리스트에서 남은 상어들의 정보를 확인한다*/
	public static void SharkLocationInfo() {
		/*상어 위치 정보: */
		System.out.println("상어 위치 정보");
		System.out.println("+-----------+----------------+------------+");
		for(int i =0; i< sharks.size() ; i++) {
			System.out.println("index "+i+" : "+ sharks.get(i).r+" , "+sharks.get(i).c);
		}
	}
/**현재 배열의 상태를 출력한다*/
	public static void MapInfo() {
		System.out.println("현재 맵: ");
		for(int i =1; i<=R; i++) {
			for(int j = 1; j<=C; j++) {
				System.out.print(arr[i][j]+" ");
			}
			System.out.println();
		}//배열 체크
		
	}
}

学んだこと