[프로그래머스] 멀쩡한 사각형 / Kotlin

https://programmers.co.kr/learn/courses/30/lessons/62048?language=kotlin 

코딩테스트 연습 - 멀쩡한 사각형

 

문제

문제 설명

가로 길이가 Wcm, 세로 길이가 Hcm인 직사각형 종이가 있습니다. 종이에는 가로, 세로 방향과 평행하게 격자 형태로 선이 그어져 있으며, 모든 격자칸은 1cm x 1cm 크기입니다. 이 종이를 격자 선을 따라 1cm × 1cm의 정사각형으로 잘라 사용할 예정이었는데, 누군가가 이 종이를 대각선 꼭지점 2개를 잇는 방향으로 잘라 놓았습니다. 그러므로 현재 직사각형 종이는 크기가 같은 직각삼각형 2개로 나누어진 상태입니다. 새로운 종이를 구할 수 없는 상태이기 때문에, 이 종이에서 원래 종이의 가로, 세로 방향과 평행하게 1cm × 1cm로 잘라 사용할 수 있는 만큼만 사용하기로 하였습니다.
가로의 길이 W와 세로의 길이 H가 주어질 때, 사용할 수 있는 정사각형의 개수를 구하는 solution 함수를 완성해 주세요.

제한사항

• W, H : 1억 이하의 자연수

입출력 예

W	H	result
8	12	80

입출력 예 설명

입출력 예 #1
가로가 8, 세로가 12인 직사각형을 대각선 방향으로 자르면 총 16개 정사각형을 사용할 수 없게 됩니다. 원래 직사각형에서는 96개의 정사각형을 만들 수 있었으므로, 96 - 16 = 80 을 반환합니다.

---

 

의식의 흐름

1. 일단 총 사각형 갯수에서 선이 지나가는 갯수를 빼야 한다는 것은 알겠다.

2. 아무리 생각해도 뭔가 공식이 될만한 게 떠오르지 않음; 이건 유형을 생각하고 외우면 좋을 것 같아서 고민하다가...

3. 3x2의 경우 6-4니까 뭔가 있을 것 같아서 계속 고민하다가 ㅋㅋㅋ

4. 결국 아래 내용을 참고해서 풀었다. 설명도 잘 되어있고 정말 똑똑하신 듯!

[프로그래머스] 멀쩡한 사각형 in python

 

해결 코드

import kotlin.math.*

class Solution {
    fun solution(w: Int, h: Int): Long {
        return (w * h - (w + h - gcd(w, h))).toLong()
    }

    fun gcd(w: Int, h: Int): Int {
        if (min(w, h) == 0) return max(w, h)
        else return gcd(max(w, h) % min(w, h), min(w, h))
    }
}

는 왜인지 모르겠지만 테스트 12~17이 fail이 뜸

은 Long() 처리를 했음에도 계속 fail이 나서...

 

다른 방법으로 풀어보기로 했다! y=(h/w)x의 그래프를 생각해서 이용해보자. 반 나눠 접고 카운트 후 *2 적용

class Solution {
    fun solution(w: Int, h: Int): Long {

        var answer = 0L

        (0 until w).forEach {
            answer += h.toLong() * it / w.toLong()
        }

        return answer * 2
    }
}