28102023-01-26 13:34:13bzsofiaÚtadócpp11Wrong answer 25/50104ms15512 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 + (el[i].db * ar[i])) % mod;
    }

    cout << sum%mod<<"\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/22ms2548 KiB
4Partially correct1/22ms2556 KiB
5Partially correct1/22ms2692 KiB
6Partially correct1/22ms3020 KiB
7Partially correct1/22ms3112 KiB
8Partially correct4/8100ms15512 KiB
9Partially correct1/24ms3256 KiB
10Partially correct1/24ms3200 KiB
11Partially correct1/24ms3196 KiB
12Partially correct1/24ms3524 KiB
13Partially correct1/24ms3808 KiB
14Partially correct1/2101ms11112 KiB
15Partially correct1/296ms11340 KiB
16Partially correct1/296ms11372 KiB
17Partially correct1/2103ms11564 KiB
18Partially correct1/2104ms11568 KiB
19Partially correct1/298ms11696 KiB
20Partially correct1/2100ms11644 KiB
21Partially correct1/297ms11756 KiB
22Partially correct1/2104ms11748 KiB
23Partially correct1/297ms11752 KiB
24Partially correct1/298ms11776 KiB