[SWEA] 무선충전 (Java 자바)

문제

https://www.swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWXRDL1aeugDFAUo

SW Expert Academy

 

 

풀이 및 소스코드

 

여기서 중요한 점은 AP의 좌표가 행-열 로 주어지는게 아니라 열-행 으로 주어진다 ㅠ 흑흑 ...!

 

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

public class Solution {
	static int m,a;
	static int[] dx = { 0, -1, 0, 1, 0 };
	static int[] dy = { 0, 0, 1, 0, -1 };
	static int[] moving_a;
	static int[] moving_b;
	static ArrayList<BC> ap; //bc 정보

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		StringTokenizer st;
		
		int t = Integer.parseInt(br.readLine());
		for (int tc = 1; tc <= t; tc++) {
			st = new StringTokenizer(br.readLine());
			m = Integer.parseInt(st.nextToken());
			a = Integer.parseInt(st.nextToken());
			int res = 0;

			moving_a = new int[m]; //a의 움직임을 저장할 배열
			moving_b = new int[m]; //b의 움직임을 저장할 배열
			
			XY A = new XY(0, 0); 
			XY B = new XY(9, 9);
			st = new StringTokenizer(br.readLine());
			for (int i = 0; i < m; i++) {
				moving_a[i] = Integer.parseInt(st.nextToken());
			}
			
			st = new StringTokenizer(br.readLine());
			for (int i = 0; i < m; i++) {
				moving_b[i] = Integer.parseInt(st.nextToken());
			}
			
			ap = new ArrayList<>();
			for (int i = 0; i < a; i++) {
				
				st = new StringTokenizer(br.readLine());
				int y = Integer.parseInt(st.nextToken()) - 1; //1부터 시작하니 1을 빼주는게 편하다.
				int x = Integer.parseInt(st.nextToken()) - 1;
				int c = Integer.parseInt(st.nextToken());
				int p = Integer.parseInt(st.nextToken());
				ap.add(new BC(x,y, c,p));
			}
			
			res += find_maxsum(A, B); //0초일 때
			for(int i=0;i<m;i++) {
				A.x = A.x+dx[moving_a[i]]; 
				A.y = A.y+dy[moving_a[i]];
				B.x = B.x+dx[moving_b[i]];
				B.y = B.y+dy[moving_b[i]];
				res += find_maxsum(A, B);
			}
			sb.append("#").append(tc).append(" ").append(res).append("\n");
		}
		System.out.println(sb);
	}

	
	public static int find_maxsum(XY aa, XY bb) {
		int[][] sum_info = new int[2][a]; //각각 bc에 대해 합계 구하기 0:a 1:b
        //sum_info[i][j] -> i가 0이면 a와, 1이면 b와 j번째 BC가 범위 안에 속한다면 j번째 BC의 p값을 넣어줌
		for(int i=0;i<a;i++) {
			
			if((Math.abs(ap.get(i).x-aa.x)+Math.abs(ap.get(i).y-aa.y))<=ap.get(i).c) {
            //a가 유효 거리내에 있다면,
				sum_info[0][i] = ap.get(i).p;
			}
			
			if((Math.abs(ap.get(i).x-bb.x)+Math.abs(ap.get(i).y-bb.y))<=ap.get(i).c) {
				sum_info[1][i] = ap.get(i).p;
			}
		}
		int max = 0;
		for(int i=0;i<a;i++) {
			for(int j=0;j<a;j++) {
				int now_sum = 0;
				if(i==j&&sum_info[0][i]==sum_info[1][j]) { //같은 bc 사용 시 하나만 해준다.
					now_sum = sum_info[0][i];
				}
				else {
					now_sum = sum_info[0][i]+sum_info[1][j];
				}
				max = Math.max(max, now_sum);
			}
		}
		return max;
	}
}
class BC { //AP 좌표와 c, p 의 값을 저장
	int x;
	int y;
	int c;
	int p;

	BC(int x, int y, int c, int p) {
		this.x = x;
		this.y = y;
		this.c = c;
		this.p = p;
	}
}


class XY {//좌표를 저장하기 위한 클래스
	int x;
	int y;

	public XY(int x, int y) {
		this.x = x;
		this.y = y;
	}

}