#include <bits/stdc++.h>
#define REP(i,n) for(int _n=(n), i=0;i<_n;++i)
#define FOR(i,a,b) for(int i=(a),_b=(b);i<=_b;++i)
#define FORD(i,a,b) for(int i=(a),_b=(b);i>=_b;--i)
using std::vector;
void init_io() {
std::cin.tie(nullptr);
std::ios::sync_with_stdio(false);
}
struct Vertex {
// [era][dir]
// no child = self loop
int children[2][2] = {};
int flatten_delta = 0;
};
int num_vertices;
vector<Vertex> vertices;
void read_input() {
std::cin >> num_vertices;
++num_vertices;
vertices.resize(num_vertices);
REP(i,num_vertices) REP(era,2) REP(dir,2) {
vertices[i].children[era][dir] = i;
}
REP(era, 2) {
FOR(i,1,num_vertices-1) {
int parent;
std::cin >> parent;
if (parent == -1) parent=0;
vertices[parent].children[era][i > parent] = i;
}
}
}
int64_t solve() {
for (const Vertex &vertex : vertices) {
REP(dir, 2) {
int a = vertex.children[0][dir];
int b = vertex.children[1][dir];
if (a > b) std::swap(a, b);
vertices[a].flatten_delta += 1;
vertices[b].flatten_delta -= 1;
}
}
int64_t result = 0;
int flatten = 0;
REP(i, num_vertices) {
const Vertex &vertex = vertices[i];
REP(dir, 2) {
if (dir==1) flatten += vertices[i].flatten_delta;
if (flatten) {
REP(era, 2) {
result += std::max(0, std::abs(vertex.children[era][dir] - i) - 1);
}
} else {
result += std::abs(vertex.children[0][dir] - vertex.children[1][dir]);
}
}
}
return result;
}
int main() {
init_io();
read_input();
int64_t res = solve();
res %= 1'000'000'007;
std::cout << res << "\n";
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | #include <bits/stdc++.h> #define REP(i,n) for(int _n=(n), i=0;i<_n;++i) #define FOR(i,a,b) for(int i=(a),_b=(b);i<=_b;++i) #define FORD(i,a,b) for(int i=(a),_b=(b);i>=_b;--i) using std::vector; void init_io() { std::cin.tie(nullptr); std::ios::sync_with_stdio(false); } struct Vertex { // [era][dir] // no child = self loop int children[2][2] = {}; int flatten_delta = 0; }; int num_vertices; vector<Vertex> vertices; void read_input() { std::cin >> num_vertices; ++num_vertices; vertices.resize(num_vertices); REP(i,num_vertices) REP(era,2) REP(dir,2) { vertices[i].children[era][dir] = i; } REP(era, 2) { FOR(i,1,num_vertices-1) { int parent; std::cin >> parent; if (parent == -1) parent=0; vertices[parent].children[era][i > parent] = i; } } } int64_t solve() { for (const Vertex &vertex : vertices) { REP(dir, 2) { int a = vertex.children[0][dir]; int b = vertex.children[1][dir]; if (a > b) std::swap(a, b); vertices[a].flatten_delta += 1; vertices[b].flatten_delta -= 1; } } int64_t result = 0; int flatten = 0; REP(i, num_vertices) { const Vertex &vertex = vertices[i]; REP(dir, 2) { if (dir==1) flatten += vertices[i].flatten_delta; if (flatten) { REP(era, 2) { result += std::max(0, std::abs(vertex.children[era][dir] - i) - 1); } } else { result += std::abs(vertex.children[0][dir] - vertex.children[1][dir]); } } } return result; } int main() { init_io(); read_input(); int64_t res = solve(); res %= 1'000'000'007; std::cout << res << "\n"; } |
English