[자바] 백준 14721 - 성적표

문제 : boj14721

 

  뭔가 무서워 보이는 공식이 있지만, 그냥 문제에서 제시된 대로 구현만 해보면 된다.

a와 b가 100 이하이므로 a를 1~100, b를 1~100 범위 내에서 모든 쌍을 확인하면서 rss를 계산하면 된다.

즉, 입력으로 x와 y가 주어지니, 모든 a,b 쌍에 대해 (y-(a*x+b))^2의 합의 최소치를 찾으면 된다.

 

코드 : github

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

public class Main {
    private void solution() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] arr = new int[n][2];
        for (int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            arr[i][0] = Integer.parseInt(st.nextToken());
            arr[i][1] = Integer.parseInt(st.nextToken());
        }
        long min = Integer.MAX_VALUE;
        int minA = 0;
        int minB = 0;
        for (int a = 1; a <= 100; a++) {
            for (int b = 1; b <= 100; b++) {
                long rss = 0;
                for (int i = 0; i < n; i++) {
                    long cur = arr[i][0]*a+b-arr[i][1];
                    rss += cur*cur;
                }
                if (min>rss) {
                    min = rss;
                    minA = a;
                    minB = b;
                }
            }
        }
        System.out.println(minA + " " + minB);
    }

    public static void main(String[] args) throws Exception {
        new Main().solution();
    }
}