Submission #533561

#	Time	Username	Problem	Language	Result	Execution time	Memory
533561		Slavita	Šarenlist (COCI22_sarenlist)	C++14	0 / 110	4 ms	4940 KiB

Main

#include <bits/stdc++.h>
/*#pragma optimize("O3")
#pragma GCC optimize("-O3")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC optimize("fast-math")*/
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define ve vector<int>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define pi pair<int,int>
#define all(v) v.begin(),v.end()
#define si(v) (int)v.size()
#define en '\n'
#define ordered_set tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
#define ordered_muiltiset tree<int, null_type,less_equal<>, rb_tree_tag,tree_order_statistics_node_update>
//#define int long long
using namespace std;
//using namespace __gnu_pbds;
typedef long long ll;
typedef unsigned long long ull;

const int N = 2e5 + 228;
const int big = 1e9 + 228;
const ll llbig = 1e18 + 228;
const ll md = 1e9 + 7;
//ordered_set os; // os.order_of_key(4), (*os.find_by_order(5))
int n, m, ans, k, q, d[N];
ve g[N];

void dfs(int v, int dist, int pr){
    d[v] = min(d[v], dist);
    for (int i = 0; i < si(g[v]); i++){
        int u = g[v][i];
        if (u == pr) continue;
        dfs(u, dist + 1, v);
    }
}

//#undef int
int main(){
    //#define int long long
    iostream::sync_with_stdio(false); cin.tie(0); ios_base::sync_with_stdio(false); cout.tie(0);
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    cin >> n >> m >> k;
    for (int i = 1; i <= n - 1; i++){
        int a, b; cin >> a >> b;
        g[a].pb(b);
        g[b].pb(a);
    }

    for (int i = 1; i <= m; i++){
        int start, finish; cin >> start >> finish;
        for (int i = 1; i <= n; i++){
            d[i] = big;
        }
        dfs(start, 0, -1);
        ll ans = k;
        for (int i = 2; i <= d[finish]; i++){
            ans *= (k - 1);
            ans %= md;
        }
        cout << ans << en;
    }

    return 0;
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 15
• Subtask 3: 0 / 10
• Subtask 4: 0 / 10
• Subtask 5: 0 / 65