BOJ: 7510 고급 수학

문제

준규는 집을 짓고 있다. 준규는 모든 벽 모양을 직각 삼각형으로 만들려고 한다. 적절히 나무를 잘라 삼각형을 만들었지만, 준규는 각도를 측정할 수 있는 도구를 가지고 있지 않다. 어쩔 수 없이 줄자를 이용해 삼각형 세 변의 길이를 측정한 다음, 직각 삼각형인지 아닌지를 알아보려고 한다.

삼각형 세 변의 길이가 주어졌을 때, 직각 삼각형인지 아닌지를 구하는 프로그램을 작성하시오.

입력

입력의 첫째 줄에는 테스트 케이스의 개수 \(n\)이 주어진다. 각 테스트 케이스는 \(1 \leq a, b, c \leq 40000\) 으로 이루어져 있다. 세 정수는 삼각형 각 변의 길이를 나타낸다.

출력

각 테스트 케이스마다 “Scenario #i:”를 출력한다. \(i\)는 테스트 케이스 번호이며, 1부터 시작한다. 그 다음 줄에는 입력으로 주어진 삼각형이 직각 삼각형인 경우에는 “yes”, 아닌 경우에는 “no”를 출력한다. 각 테스트 케이스 사이에는 빈 줄을 하나 출력한다.

예제 입력 1

2
36 77 85
40 55 69

예제 출력 1

Scenario #1:
yes

Scenario #2:
no

코드

#include <bits/stdc++.h>

using namespace std;

int main(void) {
    int n;
    
    cin>>n;
    
    for(int i=0; i<n; i++) {
        vector<int> v;
        
        for(int j=0; j<3; j++) {
            int k;
        
            cin>>k;
        
            v.push_back(k);
        }
        
        int max = *max_element(v.begin(), v.end());
        int index = max_element(v.begin(), v.end()) - v.begin();
        int sq_sum = 0;
        
        
        for(int j=0; j<3; j++) {
            if(j == index)
                continue;
            else
                sq_sum += (v[j]*v[j]);
        }
        
        cout<<"Scenario #"<<i+1<<":\n";
        
        if(sq_sum == (max * max))
            cout<<"yes\n\n";
        else
            cout<<"no\n\n";    
    }
    
    return 0;
}

Reference

BOJ