[Codeforces] 1257A: Two Rival Students

https://codeforces.com/problemset/problem/1257/A

Problem - 1257A - Codeforces

코드

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		for(int tc=0; tc<t; tc++) {
			int n = scan.nextInt();	// 학생 수
			int x = scan.nextInt();	// 스왑 수
			int a = scan.nextInt();	// 학생 1
			int b = scan.nextInt();	// 학생 2
			int max = 0;	// 두 학생사이 최대거리
			
			// swap 기회가 0이면 바로 출력
			if(x == 0) {
				System.out.println(Math.abs(a-b));
				continue;
			}
			
			// swap을 하지 않아도 이미 두학생 거리가 최대일경우도 이미 출력
			if(Math.abs(a-b) + 1 == n) {
				System.out.println(Math.abs(a-b));
				continue;
			}
			
			// 위 두경우를 제외하곤 두학생 거리를 swap횟수만큼 늘린다.
			int diff = Math.abs(a-b) + x;
			
			// 두학생 거리를 swap횟수만큼 늘렸을때, 기존의 최대범위를 넘을경우
			if(diff >= n)
				System.out.println(n-1);
			else
				System.out.println(diff);
		}
		
		scan.close();

	}

}

풀이

x 수만큼 두 학생의 거리를 벌릴 수 있고, 두 학생의 거리가 최대가 되도록 만들기.

 

1) 주어진 swap의 기회가 0인경우(x = 0) 

 - 두 학생을 이동할 수 없으므로 현재 두 학생의 거리 차이를 출력한다.

 

2) 주어진 swap의 기회와 상관없이 이미 두 학생사이의 거리가 최대일때

 - 그냥 현재 두 학생의 거리 차이가 최대이므로 그대로 출력한다.

 

3) 위 두가지가 아닌경우, swap의 기회를 기준으로 거리를 구한다.

 - 두 학생의 거리는 현재 거리의 차이에서 swap의 기회를 더하는게 최대 거리가 된다.

 - swap 한번당 두학생의 거리는 1씩 벌어지기 때문.

● 3-1) 만약 위 과정에서 swap을 모두 더했는데 두 학생의 최대 거리보다 커지는 경우 ? 

   - 더 커지면 주어진 학생수를 기준으로 최대 거리를 출력한다.(1~n이므로 거리최대는 n-1)

● 3-2) 위 과정에서 swap을 모두 더했을때 최대 거리보다 작아지는 경우는 그대로 출력한다.

 

 

 

한줄 풀이

System.out.println(Math.min(n - 1, Math.abs(a - b) + x));

위 한줄이면 끝난다 ....................... ;; wOw