취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 33164번 문제인 どら焼き (Dorayaki)이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/33164

각 도라야끼에 대하여 맛을 계산하는 데에는 \(O(1)\)의 연산이 필요하므로 모든 도라야끼의 맛을 하나하나 구하는 것은 \(O(NM)\)의 시간복잡도로 해낼 수 있다. 또한 \(NM\)은 10000 이하이므로 가능한 모든 도라야끼에 대하여 각 도라야끼의 맛을 더하는 것으로도 문제를 충분히 해결할 수 있음을 관찰할 수 있다.

위 내용을 반복문 등으로 단순히 구현하여 문제를 해결하자.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;

int N, M, ans;
int A[100], B[100];

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> N >> M;
    for (int i = 0; i < N; i++) cin >> A[i];
    for (int i = 0; i < M; i++) cin >> B[i];

    for (int i = 0; i < N; i++) {
        for (int j = 0; j < M; j++) {
            ans += (A[i] + B[j]) * max(A[i], B[j]);
        }
    }

    cout << ans;
}