8-froglike6

[froglike6]

🔗 문제 링크

곰곰이와 시소

✔️ 소요된 시간

30분

✨ 수도 코드

수도코드

함수 시작:

  입력 N, L ← 첫 번째 줄에서 읽음

  x_list ← 두 번째 줄에서 공백으로 구분된 N개의 정수 입력 받음
  w_list ← 세 번째 줄에서 공백으로 구분된 N개의 정수 입력 받음

  total_weight ← 0
  weighted_sum ← 0

  반복 i = 0부터 N - 1까지:
    total_weight ← total_weight + w_list[i]
    weighted_sum ← weighted_sum + (x_list[i] × w_list[i])

  pivot ← weighted_sum ÷ total_weight

  소수점 아래 10자리까지 pivot 출력

함수 종료

시험기간이라 간단한 문제를 들고 왔습니다(원래 계획은 볼록껍질이었습니다...).

문제에서 시소의 길이와, 그 위에 놓여있는 치킨들의 무게와 그 위치들을 줍니다. 이 입력들을 기반으로, 받침점이 어디에 있어야 수평이 되는지 출력하는 물리학 문제입니다.

저는 간단히 토크 평형을 이용해서, 식을 다음과 같이 세웠습니다.
$$\sum_{i=1}^{N} w_i \cdot (x_i - x) = 0$$

이 식을 정리하면 다음과 같이 $x$에 대해 정리가 됩니다.
$$x = \frac{\sum_{i=1}^{N} w_i x_i}{\sum_{i=1}^{N} w_i}$$

따라서 for문으로 모든 치킨들을 돌아가면서 수식을 계산하였습니다.

📚 새롭게 알게된 내용

[dohyeondol1]

문제만 읽고 식이 세워져서 바로 제출했다가 몇 번을 틀렸는지..

저는 accumulate함수를 이용해 w의 누적 합을 구하고,
도현님의 식과 같은 식으로 해결하였습니다.

중간에 자료형을 제대로 못 맞추는 바람에 몇번 틀렸네요...
accumulate함수에서도,
세 번째 인자를 0으로 설정해 놓으면 덧셈이 정수 연산이 된다는 사실을 이제야 알게 되었습니다..

C++ 코드

#include <iostream>
#include <vector>
#include <numeric>
#include <iomanip>
using namespace std;

int main() {
    cin.tie(0)->sync_with_stdio(0);
    
    int N, L;
    cin >> N >> L;
    
    vector<int> x(N);
    for(int &position : x)
        cin >> position;
    
    vector<int> w(N);
    for(int &weight : w)
        cin >> weight;
    
    double totalW = accumulate(w.begin(), w.end(), 0.0);
    double totalXTimesW = 0.0;
    for(int i = 0; i < N; i++)
        totalXTimesW += static_cast<double>(x[i]) * w[i];
    
    double result = totalXTimesW / totalW;
    
    cout << fixed << setprecision(10) << result << '\n';
    
    return 0;
}

오버플로우를 조심, 또 조심해야 한다는 걸 느꼈습니다..

[caucsejunseo]

식을 쓴 다음 x에 대하여 정리하는 방식이 수학을 풀 때는 당연하게 하는데 백준 문제를 풀 때는 어색하네요. 처음에 어떻게 해야 하나 고민했습니다...
오랜만에 물리 문제 재밌었습니다.

[hadongun]

오랜만에 식을 세우려 하니 자꾸 뇌정지가 오네요..ㅎㅎ
전체적인 코드는 비슷한데 확실히 도현님의 파이썬 코드가 깔끔하신 것 같아요!
zip함수랑 포매팅 문법도 배워 가요.!