Областная олимпиада по информатике 2020 год, 9-11 классы

Задача E. Есмахан және стартап

Ограничение по времени:

1 second

Ограничение по памяти:

256 megabytes

Есмахан - морж өсіру бойынша стартап бастады. Ол $n-1$ қызметкеріді жалдады. Есмахан - нөмірі $1$ қызметкер, қалғандар $2$-ден $n$-ге дейін нөмірленген. Әр қызметкерде бір тікелей жетекші $p_i$ бар. Есмаханның бастығы жоқ. $p_i$ мәндері данақ құрайды. Енді Есмахан оларға төлеуі керек. $i$ қызметкері $a_i$ теңге алғысы келеді. $i$ қызметкері $b_i$ теңге берілсін, сонда оның наразылығы $|a_i - b_i|$ болады. Есмаханның қағидасы бар - \textbf {әр қызметкер өзінің бағыныштыларына қарағанда көбірек алуы керек}. Жалақыны қызметкерлердің жалпы наразылық минималды болатындай етіп бөлу керек.

Формат входного файла

Бірінші жолда бір бүтін сан $n$ $(1 <= n <= 200000)$. Екінші жолда $n - 1$ бүтін сан $p_2, p_3, ... p_n$ $(1 <= p_i <= i - 1)$ - бұл бастықтардың нөмірлері. Үшінші жолда $n$ бүтін сан $a_1, a_2, ... a_n$ $(1 <= a_i <= 10^9)$ - жұмысшылардың күтуі.

Формат выходного файла

Бір бүтін санды басып шығарыңыз - қызметкерлердің жалпы наразылығы.

Система оценки

Есеп $25$ тесттен тұрады, әр тест $4$ ұпайға бағаланады.

1. Берілген тест.
2. $1 <= n <= 1000$, $1 <= a_i <= 1000$ әр $1 <= i <= n$, әр қызметкерде бағыныштылар саны бірден көп емес| 2 тест.
3. $1 <= n <= 1000$, $1 <= a_i <= 1000$ әр $1 <= i <= n$ | 2 тест.
4. $1 <= n <= 1000$, әр қызметкерде бағыныштылар саны бірден көп емес | 2 тест.
5. $1 <= n <= 1000$ | 3 тест.
6. $1 <= n <= 200000$, әр қызметкерде бағыныштылар саны бірден көп емес | 5 тест.
7. $1 <= n <= 200000$ | 10 тест.

Пример:

Вход

7
1 2 1 1 5 6
1 2 3 1 4 3 3

Ответ

7

Замечание

Мысалда $b={5, 4, 3, 1, 4, 3, 2}$ ( Batyr Sardarbekov )

Комментарий/решение

кодты корсету/жасыру

int n, a[N], mx[N], sz[N];
long long ans[N];
vector<int> g[N];
priority_queue<int> q[N];

void make(int v = 1) {
	sz[v] = 1;
	for (int to : g[v]) {
		make(to);
		sz[v] += sz[to];
		if (sz[mx[v]] < sz[to])
			mx[v] = to;
	}
}


void dfs(int v = 1, int d = 0) {
	a[v] += d;
	if (sz[v] == 1) {
		q[v].push(a[v]);
		return;
	}
	for (int to : g[v]) {
		dfs(to, d + 1);
		ans[v] += ans[to];
		if (mx[v] != to) {
			while (q[to].size()) {
				q[v].push(q[to].top());
				q[to].pop();
			}
		}
	}
	q[v].swap(q[mx[v]]);
	while (q[mx[v]].size()) {
		q[v].push(q[mx[v]].top());
		q[mx[v]].pop();
	}
	if (q[v].size() && a[v] < q[v].top())
		ans[v] += q[v].top() - a[v];
	q[v].push(a[v]);
	q[v].push(a[v]);
	q[v].pop();
}

int main() {
	speed;
	cin >> n;
	for (int i = 2, p; i <= n; ++i) {
		cin >> p;
		g[p].push_back(i);
	}
	for (int i = 1; i <= n; ++i) {
		cin >> a[i];
	}
	make();
	dfs();
	cout << ans[1] << '\n';
}

кодты корсету/жасыру

from sys import setrecursionlimit

def make(n, p):
    global graph
    for i in range(n):
        s = set()
        for j in range(n-1):
            if i == p[j]-1:
                s.add(j+1)
        graph[i] = s


def dfs(graph, start, visited=None):
    global a
    if visited is None:
        visited = set()
    visited.add(start)
    for next in graph[start] - visited:
        if a[start] < a[next]:
            a[start] = a[next] + 1
            dfs(graph, 0, visited)
        elif a[start] == a[next]:
            a[next] -= 1
            dfs(graph, next, visited)
        else:
            dfs(graph, next, visited)
    return visited
               
graph = dict()

n = int(input())
p = list(map(int ,input().split()))
a = list(map(int, input().split()))
b = a.copy()
make(n, p)
dfs(graph, 0)

cnt = 0
for i in range(n):
    cnt += abs(a[i] -b[i])

print(cnt)

setrecursionlimit(10**9)

кодты корсету/жасыру

#include <bits/stdc++.h>

//#define int long long
#define vt vector
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define sz(x) (int)(x).size()
#define ff first
#define ss second
#define dbg(x) cerr << #x << " = " << x << '\n'
#define bit(x, i) ((x) >> (i) & 1)

using namespace std;
using ll = long long;
using pii = pair<int, int>
using vvt = vt< vt<int> >

const int N = 2e5 + 5, mod = 1e9 + 7, B = 500;
const ll inf = 1e18 + 7, LIM = (1ll << 60);
const double eps = 1e-6;

int n;
ll a[N], dp[N];
vt<int> g[N];
priority_queue<ll> q[N];

void dfs (int v, int p = -1, int dist = 1) {
	a[v] += dist;
	for (auto to: g[v]) {
		if (to != p) {
			dfs(to, v, dist + 1);
			dp[v] += dp[to];
			if (sz(q[to]) > sz(q[v])) q[v].swap(q[to]);
			while (!q[to].empty()) q[v].push(q[to].top()), q[to].pop();
		}
	}
	if (!q[v].empty() && a[v] < q[v].top()) dp[v] += q[v].top() - a[v];
	q[v].push(a[v]), q[v].push(a[v]);
	q[v].pop();
}

void solve () {
	cin >> n;
	for (int i = 2, p; i <= n; i++) cin >> p, g[p].pb(i), g[i].pb(p);
	for (int i = 1; i <= n; i++) cin >> a[i];
	dfs(1);
	cout << dp[1];	
	cout << '\n';
}

bool testcases = 0;

signed main() {
	cin.tie(0) -> sync_with_stdio(0);
	int test = 1;
	if (testcases) cin >> test;
	for (int cs = 1; cs <= test; cs++) {
		solve();
	}
}