Submission #381973

#	Time	Username	Problem	Language	Result	Execution time	Memory
381973		Slavita	Janjetina (COCI21_janjetina)	C++14	15 / 110	1576 ms	6636 KiB

Main

#include <bits/stdc++.h>
#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'
using namespace std;
typedef long long ll;
typedef unsigned long long ull;

const int N = 1e5 + 228;
const int big = 1e9;
int n, k, ans;
bool mrk[N];
vector<pi> g[N];

void dfs(int v, int st, int mxn){
    if (st != 0 && mxn - st >= k){
        ans++;
    }
    mrk[v] = 1;
    for (int i = 0; i < si(g[v]); i++){
        int u = g[v][i].fi;
        if (mrk[u]) continue;
        dfs(u, st + 1, max(mxn, g[v][i].se));
    }
}

int main(){
    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 >> k;
    for (int i = 1; i <= n - 1; i++){
        int a, b, x;
        cin >> a >> b >> x;
        g[a].pb(mp(b, x));
        g[b].pb(mp(a, x));
    }

    ans = 0;
    for (int i = 1; i <= n; i++){
        //if (i == j) continue;
        for (int i = 1; i <= n; i++) mrk[i] = 0;
        dfs(i, 0, 0);
        //cout << i << '|' << j << '|' << dans << "|k=" << k << en;
    }
    cout << ans;
    return 0;
}

• Subtask 1: 15 / 15
• Subtask 2: 0 / 35
• Subtask 3: 0 / 60