[11265] 끝나지 않는 파티

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

11265번: 끝나지 않는 파티

<문제>

입력받은 인접행렬로 플로이드를 돌려두고, 쿼리당 O(1)에 해결하는 문제

 

<소스코드>

#include <bits/stdc++.h>

using namespace std;

using ll = long long;

const ll INF = (ll)1e13;

ll n, m, a[501][501];

int main(void) {

    ios_base::sync_with_stdio(0);

    cin.tie(0);

    cin >> n >> m;

    ll i, j, k;

    for (i = 1; i <= n; i++)

        for (j = 1; j <= n; j++) cin >> a[i][j];

    for (k = 1; k <= n; k++)

        for (i = 1; i <= n; i++)

            for (j = 1; j <= n; j++) a[i][j] = min(a[i][k] + a[k][j], a[i][j]);

    while (m--) {

        ll A, B, C;

        cin >> A >> B >> C;

        (a[A][B] > C) ? cout << "Stay here\n" : cout << "Enjoy other party\n";

    }

    return 0;

}