#include <bits/stdc++.h> using namespace std; #define F first #define S second #define pb push_back #define pii pair < int, int > #define ll long long int sz[200001]; vector < pii > g[200001]; bool used[200001]; void calc(int v, int pr) { sz[v] = 1; for (auto to:g[v]) { if (to.F == pr || used[to.F] == 1) continue; calc(to.F, v); sz[v] += sz[to.F]; } } int32_t main() { int n; cin >> n; int x[n], y[n], z[n]; for (int i = 0;i < n - 1;i++) { cin >> x[i] >> y[i] >> z[i]; x[i]--;y[i]--; g[x[i]].pb({y[i], z[i]}); g[y[i]].pb({x[i], z[i]}); } calc(0, 0); int ans = 0; for (int i = 0;i < n - 1;i++) if (sz[x[i]] > sz[y[i]]) ans += z[i] * sz[y[i]] * (n - sz[y[i]]); else ans += z[i] * sz[x[i]] * (n - sz[x[i]]); cout << ans; }
Write, Run & Share C++ code online using OneCompiler's C++ online compiler for free. It's one of the robust, feature-rich online compilers for C++ language, running on the latest version 17. Getting started with the OneCompiler's C++ compiler is simple and pretty fast. The editor shows sample boilerplate code when you choose language as C++
and start coding!
OneCompiler's C++ online compiler supports stdin and users can give inputs to programs using the STDIN textbox under the I/O tab. Following is a sample program which takes name as input and print your name with hello.
#include <iostream>
#include <string>
using namespace std;
int main()
{
string name;
cout << "Enter name:";
getline (cin, name);
cout << "Hello " << name;
return 0;
}
C++ is a widely used middle-level programming language.
When ever you want to perform a set of operations based on a condition If-Else is used.
if(conditional-expression) {
//code
}
else {
//code
}
You can also use if-else for nested Ifs and If-Else-If ladder when multiple conditions are to be performed on a single variable.
Switch is an alternative to If-Else-If ladder.
switch(conditional-expression){
case value1:
// code
break; // optional
case value2:
// code
break; // optional
......
default:
code to be executed when all the above cases are not matched;
}
For loop is used to iterate a set of statements based on a condition.
for(Initialization; Condition; Increment/decrement){
//code
}
While is also used to iterate a set of statements based on a condition. Usually while is preferred when number of iterations are not known in advance.
while (condition) {
// code
}
Do-while is also used to iterate a set of statements based on a condition. It is mostly used when you need to execute the statements atleast once.
do {
// code
} while (condition);
Function is a sub-routine which contains set of statements. Usually functions are written when multiple calls are required to same set of statements which increases re-usuability and modularity. Function gets run only when it is called.
return_type function_name(parameters);
function_name (parameters)
return_type function_name(parameters) {
// code
}