| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 447853 | MilosMilutinovic | 친구 (IOI14_friend) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#define ll long long
#define mp make_pair
#define fi first
#define se second
#define pb push_back
#define vi vector<int>
#define pi pair<int, int>
#define mod 1000000007
template<typename T> bool chkmin(T &a, T b){return (b < a) ? a = b, 1 : 0;}
template<typename T> bool chkmax(T &a, T b){return (b > a) ? a = b, 1 : 0;}
ll ksm(ll a, ll b) {if (b == 0) return 1; ll ns = ksm(a, b >> 1); ns = ns * ns % mod; if (b & 1) ns = ns * a % mod; return ns;}
using namespace std;
const int maxn = 1050;
vi g[maxn];
ll dp[maxn][2];
int a[maxn];
bool was[maxn];
bool sub2(vector<int> a) {
for (int x : a) if (x != 1) return false;
return true;
}
bool sub3(vector<int> a) {
for (int x : a) if (x != 2) return false;
return true;
}
bool sub4(vector<int> a) {
for (int x : a) if (x != 0) return false;
return true;
}
void dfs(int u, int p) {
vector<ll> s(2);
for (int v : g[u]) {
if (v != p) {
dfs(v, u);
s[0] += max(dp[v][0], dp[v][1]);
s[1] += dp[v][0];
}
}
dp[u][1] = a[u] + s[0];
dp[u][0] = s[1];
}
ll findSample(int n, vi confidence, vi host, vi protocol) {
if (sub2(protocol)) {
ll sum = accumulate(confidence.begin(), confidence.end(), 0);
return sum;
}
if (sub3(protocol)) {
return *max_element(confidence.begin(), confidence.end());
}
if (sub4(protocol)) {
for (int i = 1; i < n; i++) g[host[i]].pb(i);
for (int i = 0; i < n; i++) a[i] = confidence[i];
ll res = 0;
for (int i = 0; i < n; i++) {
if (!was[i]) {
dfs(i, i);
res += max(dp[i][0], dp[i][1]);
}
}
return res;
}
}