28082023-01-26 13:32:55bzsofiaÚtadócpp11Wrong answer 25/50101ms15376 KiB
// Utado.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <iostream>
#include <vector>
#include <map>
#include <queue>
#include <algorithm>

using namespace std;

struct adat
{
    bool lat;
    int ag = 1;
    vector <int> sz;
};

struct adat2
{
    int db;
    pair<int, int> par;
};

int i, n, a, b, mod=32609;
long long sum;
vector <adat> x;

void mely(int akt)
{
    x[akt].lat = 1;
    for (auto& e : x[akt].sz)
    {
        if (!x[e].lat)
        {
            mely(e);
            x[akt].ag += x[e].ag;
        }
    }
}

bool comp(adat2 a, adat2 b)
{
    return a.db < b.db;
}

int main()
{
    cin >> n;
    vector <adat2> el(n);
    vector <int> ar(n);
    x.resize(n + 1);

    for (i = 1; i < n; ++i)
    {
        cin >> a >> b;

        el[i].par = { a, b };
        x[a].sz.push_back(b);
        x[b].sz.push_back(a);
    }

    for (i = 1; i < n; ++i)
    {
        cin >> ar[i];
    }

    mely(1);

    sort(ar.begin() + 1, ar.end());

    for (i = 1; i < n; ++i)
    {
        b = el[i].par.second;
        el[i].db = 2 * x[b].ag * (n - x[b].ag);
        //cout << el[i].par.first << " " << el[i].par.second << " " << el[i].db << "\n";
    }

    sort(el.begin() + 1, el.end(), comp);

    for (i = 1; i < n; ++i)
    {
        sum = sum % mod + (el[i].db * ar[i]) % mod;
    }

    cout << sum<<"\n";
    for (i = 1; i < n; ++i)
    {
        cout << el[i].par.first << " " << el[i].par.second << " " << ar[i] << "\n";
    }

    return 0;
}

// Run program: Ctrl + F5 or Debug > Start Without Debugging menu
// Debug program: F5 or Debug > Start Debugging menu

// Tips for Getting Started: 
//   1. Use the Solution Explorer window to add/manage files
//   2. Use the Team Explorer window to connect to source control
//   3. Use the Output window to see build output and other messages
//   4. Use the Error List window to view errors
//   5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project
//   6. In the future, to open this project again, go to File > Open > Project and select the .sln file
SubtaskSumTestVerdictTimeMemory
base25/50
1Accepted0/03ms1808 KiB
2Wrong answer0/039ms5100 KiB
3Partially correct1/22ms2144 KiB
4Partially correct1/22ms2272 KiB
5Partially correct1/22ms2348 KiB
6Partially correct1/22ms2476 KiB
7Partially correct1/22ms2620 KiB
8Partially correct4/8101ms15376 KiB
9Partially correct1/24ms3264 KiB
10Partially correct1/24ms3520 KiB
11Partially correct1/24ms3732 KiB
12Partially correct1/24ms3680 KiB
13Partially correct1/24ms3616 KiB
14Partially correct1/298ms11356 KiB
15Partially correct1/296ms11444 KiB
16Partially correct1/294ms11456 KiB
17Partially correct1/294ms11424 KiB
18Partially correct1/294ms11400 KiB
19Partially correct1/294ms11404 KiB
20Partially correct1/297ms11524 KiB
21Partially correct1/297ms11656 KiB
22Partially correct1/297ms11860 KiB
23Partially correct1/297ms11944 KiB
24Partially correct1/297ms12056 KiB