oj.uz

Submission #1046067

# Submission time Handle Problem Language Result Execution time Memory
1046067 2024-08-06T09:39:29 Z Mher777 Petrol stations (CEOI24_stations) C++17
18 / 100
 38 ms 16684 KB 
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <iomanip>
#include <array>
#include <string>
#include <algorithm>
#include <cmath>
#include <set>
#include <map>
#include <unordered_set>
#include <unordered_map>
#include <vector>
#include <stack>
#include <queue>
#include <deque>
#include <bitset>
#include <list>
#include <iterator>
#include <numeric>
#include <complex>
#include <utility>
#include <random>
#include <cassert>
#include <fstream>
using namespace std;
mt19937_64 rnd(7069);

/* -------------------- Typedefs -------------------- */

typedef int itn;
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
typedef float fl;
typedef long double ld;

/* -------------------- Usings -------------------- */

using vi = vector<int>;
using vll = vector<ll>;
using mii = map<int, int>;
using mll = map<ll, ll>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

/* -------------------- Defines -------------------- */

#define ff first
#define ss second
#define pub push_back
#define pob pop_back
#define puf push_front
#define pof pop_front
#define mpr make_pair
#define yes cout<<"YES\n"
#define no cout<<"NO\n"
#define all(x) (x).begin(), (x).end()
#define USACO freopen("feast.in", "r", stdin); freopen("feast.out", "w", stdout);

/* -------------------- Constants -------------------- */

const int dx[8] = { -1, 0, 1, 0, -1, -1, 1, 1 };
const int dy[8] = { 0, -1, 0, 1, -1, 1, -1, 1 };
const int MAX = int(1e9 + 5);
const ll MAXL = ll(1e18) + 5ll;
const ll MOD = ll(1000000007);
const ll MOD2 = ll(998244353);

/* -------------------- Functions -------------------- */

void fastio() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
}

void precision(int x) {
    cout.setf(ios::fixed | ios::showpoint);
    cout.precision(x);
}

ll gcd(ll a, ll b) {
    if (a == 0 || b == 0) return(max(a, b));
    while (b) {
        a %= b;
        swap(a, b);
    }
    return a;
}

ll lcm(ll a, ll b) {
    return a / gcd(a, b) * b;
}

ll range_sum(ll a, ll b) {
    if (a > b) return 0ll;
    ll dif = a - 1, cnt = b - a + 1;
    ll ans = ((b - a + 1) * (b - a + 2)) / 2;
    ans += ((b - a + 1) * dif);
    return ans;
}

string dec_to_bin(ll a) {
    string s = "";
    for (ll i = a; i > 0; ) {
        ll k = i % 2;
        i /= 2;
        char c = k + 48;
        s += c;
    }
    if (a == 0) {
        s = "0";
    }
    reverse(all(s));
    return s;
}

ll bin_to_dec(string s) {
    ll num = 0;
    for (int i = 0; i < s.size(); i++) {
        num *= 2ll;
        num += (s[i] - '0');
    }
    return num;
}

ll factorial_by_mod(ll n, ll mod) {
    ll ans = 1;
    ll num;
    for (ll i = 1; i <= n; ++i) {
        num = i % mod;
        ans *= num;
        ans %= mod;
    }
    return ans;
}

bool isPrime(ll a) {
    if (a == 1) return false;
    for (ll i = 2; i * i <= a; i++) {
        if (a % i == 0) return false;
    }
    return true;
}

ll binpow(ll a, ll b) {
    if (!a) return 0;
    ll ans = 1;
    while (b) {
        if (b & 1) {
            ans *= a;
        }
        b >>= 1;
        a *= a;
    }
    return ans;
}

ll binpow_by_mod(ll a, ll b, ll mod) {
    if (!a) return 0;
    ll ans = 1;
    while (b) {
        if (b & 1) {
            ans *= a;
            ans %= mod;
        }
        b >>= 1;
        a *= a;
        a %= mod;
    }
    return ans;
}

/* -------------------- Solution -------------------- */

const int N = 70005;
int used[N];
ll ans[N], cnt[N], sub[N], pref[N], edge[N];
vector<pair<int, ll>> g[N];
vi v;
int n, m = 0;
ll k;

void dfs(int u) {
    used[u] = 1;
    cnt[u] = 1;
    v.pub(u);
    for (auto to : g[u]) {
        if (used[to.ff]) continue;
        edge[++m] = to.ss;
        pref[to.ff] = pref[u] + to.ss;
        dfs(to.ff);
        cnt[u] += cnt[to.ff];
    }
}

void slv() {
    cin >> n >> k;
    for (int i = 0; i < n - 1; ++i) {
        int x, y;
        ll c;
        cin >> x >> y >> c;
        ++x, ++y;
        g[x].pub({ y,c });
        g[y].pub({ x,c });
    }
    int node1 = 0, node2 = 0;
    for (int i = 1; i <= n; ++i) {
        if ((int)g[i].size() == 1) {
            if (!node1) node1 = i;
            else node2 = i;
        }
    }
    dfs(node1);
    for (int i = 0; i < n; ++i) {
        ++sub[v[i]];
        int node = v[i];
        int l = i + 1, r = n - 2, mid, ind = -1;
        while (l <= r) {
            mid = (l + r) / 2;
            if ((k - (pref[v[mid]] - pref[node])) < edge[mid + 1]) {
                ind = v[mid];
                r = mid - 1;
            }
            else {
                l = mid + 1;
            }
        }
        if (ind != -1) {
            ans[ind] += sub[node] * (cnt[ind] - 1ll);
            sub[ind] += sub[node];
        }
    }
    m = 0;
    v.clear();
    for (int i = 1; i <= n; ++i) {
        sub[i] = pref[i] = used[i] = cnt[i] = 0;
    }
    dfs(node2);
    for (int i = 0; i < n; ++i) {
        ++sub[v[i]];
        int node = v[i];
        int l = i + 1, r = n - 2, mid, ind = -1;
        while (l <= r) {
            mid = (l + r) / 2;
            if ((k - (pref[v[mid]] - pref[node])) < edge[mid + 1]) {
                ind = v[mid];
                r = mid - 1;
            }
            else {
                l = mid + 1;
            }
        }
        if (ind != -1) {
            ans[ind] += sub[node] * (cnt[ind] - 1ll);
            sub[ind] += sub[node];
        }
    }
    for (int i = 1; i <= n; ++i) {
        cout << ans[i] << '\n';
    }
}

void cs() {
    int tstc = 1;
    //cin >> tstc;
    while (tstc--) {
        slv();
    }
}

void precalc() {
    return;
}

int main() {
    fastio();
    precalc();
    //precision(0);
    cs();
    return 0;
}

Compilation message

Main.cpp: In function 'll range_sum(ll, ll)':
Main.cpp:97:21: warning: unused variable 'cnt' [-Wunused-variable]
   97 |     ll dif = a - 1, cnt = b - a + 1;
      |                     ^~~
Main.cpp: In function 'll bin_to_dec(std::string)':
Main.cpp:120:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  120 |     for (int i = 0; i < s.size(); i++) {
      |                     ~~^~~~~~~~~~

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 0 ms 4700 KB Output is correct

Subtask #1
0.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 0 ms 4700 KB Output is correct
3 Incorrect 1 ms 4700 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 33 ms 15192 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 0 ms 4700 KB Output is correct
3 Correct 1 ms 4700 KB Output is correct
4 Correct 33 ms 15192 KB Output is correct
5 Correct 38 ms 16684 KB Output is correct
6 Correct 34 ms 16076 KB Output is correct

Subtask #4
0.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 0 ms 4700 KB Output is correct
3 Incorrect 33 ms 9280 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #5
0.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 0 ms 4700 KB Output is correct
3 Incorrect 33 ms 9280 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #6
0.0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 4700 KB Output is correct
2 Correct 0 ms 4700 KB Output is correct
3 Incorrect 1 ms 4700 KB Output isn't correct
4 Halted 0 ms 0 KB -