[CodeTree] 싸움땅 - Java

✨ 구현

https://www.codetree.ai/training-field/frequent-problems/problems/battle-ground/description?page=1&pageSize=20

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

---

📍 고려해야할 점 

• 해당하는 칸에 있는 총들을 비교하고 가장 공격력이 높은 총을 선택해야한다.
• 진 사람의 경우 총을 해당하는 칸에 두고, 조건에 따라 이동한다.

 

🕹️ 알고리즘 과정 

진 사람이 총을 버리고 이동해야하는데, 버리는 코드를 작성하지 않아 디버깅하는데 시간을 생각보다 많이 쏟은 것 외에는 딱히 막히는 부분이 없었다.

그냥 문제 읽고 적당한 자료구조를 선택해서 문제에서 요구한대로 잘 풀면 된다. 

 

 

 

 

 

👩‍💻전체 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.StringTokenizer;

public class final_2022_1_싸움땅 {
	
	static int n, m, k;
	static int[] dx = {-1, 0, 1, 0};
	static int[] dy = {0, 1, 0, -1};
	static int[][] gunPower;
	static int[][] playerLoc;
	static Info[] players;
	static int[] point;
	static int winner, loser;
	static ArrayList<Integer>[][] guns;
	
	static class Info{
		int x, y, d, s, gp, n;
		
		public Info(int x, int y, int d, int s, int gp, int n) {
			this.x=x;
			this.y=y;
			this.d=d;
			this.s=s; // 초기 능력치 
			this.gp=gp; // 현재 가지고 있는 총의 공격력 
			this.n=n; // 플레이어 번호 
		}
	}
	
	static void simulate() {
		move();
	}
	
	static void move() {
		for(int i=1; i<=m; i++) {
			Info now = players[i];
			
			int nx = now.x+dx[now.d];
			int ny = now.y+dy[now.d];
			
			// 격자를 벗어나는 경우에는 정반대 방향으로 방향 바꾸고 1칸 이동 
			if(nx<0 || nx>=n || ny<0 || ny>=n) {
				if(now.d==0 || now.d==1) now.d+=2;
				else now.d-=2;
				
				nx = now.x+dx[now.d];
				ny = now.y+dy[now.d];
			}
			// 이동 완료 
			now.x=nx;
			now.y=ny;
			
			
			// 만약 이동한 방향에 플레이어가 없다면 총 있는지 확인 
			if(playerLoc[now.x][now.y]==0) {
				// 총이 있다면 
				if(guns[now.x][now.y].size()>=1) {
					compareGun(now);
				}
				
				// 이동한 것 playerLoc에 반영시키기 
				copyLoc();
			}
			// 만약 플레이어가 있다면 싸움 
			else {
				int idx = playerLoc[nx][ny];
				Info p = players[idx];
				fight(p, now);
				
				// 이긴 사람 & 진 사람 이동 
				Info w = players[winner];
				Info l = players[loser];
				w.x= nx;
				w.y= ny;
				l.x=nx;
				l.y=ny;
				copyLoc();
				
				// 진 사람 --> 본인이 가지고 있는 총을 해당 격자에 내려놓고 조건에 맞게 이동 
				lose(l);
				// 이긴 사람 --> 공격력 높은 총 선택 
				compareGun(w);
				copyLoc();
			}
		}
	}
	
	static void compareGun(Info p) {
 		int playerGp= p.gp;
		
		for(int i=0; i<guns[p.x][p.y].size(); i++) {
			if(playerGp<=guns[p.x][p.y].get(i)) {
				playerGp = guns[p.x][p.y].get(i);
			}
		}
		if(p.gp!=0) guns[p.x][p.y].add(p.gp);
		guns[p.x][p.y].remove(Integer.valueOf(playerGp));
		p.gp = playerGp;
	}
	
	static void fight(Info other, Info now) {
		if(other.gp+other.s == now.gp+now.s) {
			if(other.s<now.s) {
				winner = now.n;
				loser = other.n;
			}
			else if(other.s> now.s) {
				winner = other.n;
				loser = now.n;
			}
		}else if(other.gp+other.s > now.gp+now.s) {
			winner=other.n;
			loser = now.n;
			point[winner]+= ((other.gp+other.s) - (now.gp+now.s));
		}
		else {
			winner = now.n;
			loser = other.n;
			point[winner]+= ((now.gp+now.s) - (other.gp+other.s));
		}
	}
	
	static void lose(Info l) {
		
		// 본인이 가지고 있는 총을 해당 격자에 내려놓음
		if(l.gp!=0) {
			guns[l.x][l.y].add(l.gp);
			l.gp=0;
		}
		
		
		while(true) {
			
			// 플레이어가 원래 가지고 있던 방향대로 한 칸 이동
			int nx = l.x+dx[l.d];
			int ny = l.y+dy[l.d];
			
			if(nx<0 || nx>=n || ny<0 || ny>=n || playerLoc[nx][ny]!=0) {
				l.d = (l.d+1)%4;
			}else {
				l.x=nx;
				l.y=ny;
				break;
			}
		}
		compareGun(l);
	}
	
	static void copyLoc() {
		playerLoc = new int[n][n];
		
		for(int i=1; i<=m; i++) {
			Info p = players[i];
			playerLoc[p.x][p.y]= i; 
		}
	}
	
	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 st = new StringTokenizer(br.readLine());
		
		n = Integer.parseInt(st.nextToken());
		m = Integer.parseInt(st.nextToken());
		k = Integer.parseInt(st.nextToken());
		
		gunPower = new int[n][n];
		playerLoc = new int[n][n];
		players = new Info[m+1];
		point = new int[m+1];
		guns = new ArrayList[n][n];
		
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				guns[i][j] = new ArrayList<>();
			}
		}
		
		for(int i=0; i<n; i++) {
			st = new StringTokenizer(br.readLine());
			for(int j=0; j<n; j++) {
				int n =Integer.parseInt(st.nextToken());
				if(n==0) continue;
				guns[i][j].add(n); // 총 공격력 
			}
		}
		
		for(int i=1; i<=m; i++) {
			st = new StringTokenizer(br.readLine());
			
			int x = Integer.parseInt(st.nextToken())-1;
			int y = Integer.parseInt(st.nextToken())-1;
			int d = Integer.parseInt(st.nextToken()); // 방향 
			int s = Integer.parseInt(st.nextToken()); // 초기 능력치 
			
			playerLoc[x][y]=i;
			players[i] = new Info(x, y, d, s, 0, i);
		}
		
		while(k-->0) {
			simulate();
		}
		
		for(int i=1; i<=m; i++) {
			bw.write(String.valueOf(point[i]) + " ");
		}
		
		bw.flush();
		bw.close();
	}
}