| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 447855 | MilosMilutinovic | 친구 (IOI14_friend) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#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, int* confidence, int* host, int* 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;
}
return 0LL;
}