loj#2509. 「AHOI / HNOI2018」排列(思维题 set)
题意
Sol
神仙题Orz
首先不难看出如果我们从\(a_i\)向\(i\)连一条边,我们会得到以\(0\)为根的树(因为每个点一定都有一个入度,出现环说明无解),同时在进行排列的时候需要保证父亲节点一定在孩子节点之前出现
接下来考虑直接贪心。对于某些权值很小的点,我们需要让其尽早出现,同时又要满足选择的条件。
那么我们可以从小的点开始,依次向他的父亲合并,并删除该点(也就是如果父亲一但被删除,那么这个点立马被删除)
下面的内容抄袭摘抄自这里
然后直接用set搞一搞
复杂度:\(O(n\log n)\)
#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int MAXN = 5e5 + 10, SS = 1e7 + 10;
template<typename A, typename B> inline void chmax(A &x, B y) {
x = x > y ? x : y;
}
inline int read() {
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
int N, a[MAXN], fa[MAXN], vis[MAXN], ufa[MAXN];
LL w[MAXN], siz[MAXN];
vector<int> v[MAXN];
struct comp {
bool operator ()(int x, int y) {
return w[x] * siz[y] == w[y] * siz[x] ? x < y : w[x] * siz[y] < w[y] * siz[x];
}
};
int find(int x) {
return ufa[x] == x ? ufa[x] : ufa[x] = find(ufa[x]);
}
set<int, comp> s;
int dfs(int x) {
vis[x] = 1;
for(auto &to : v[x]) {
if(vis[to] == 1) return 1;
if(dfs(to)) return 1;
}
vis[x] = 2;
return 0;
}
int main() {
N = read();
for(int i = 1; i <= N; i++) {
fa[i] = read();
v[fa[i]].push_back(i);
}
for(int i = 1; i <= N; i++)
if(!vis[i])
if(dfs(i)) {puts("-1"); return 0;}
for(int i = 1; i <= N; i++) w[i] = read(), ufa[i] = i, siz[i] = 1, s.insert(i);
siz[0] = 1; ufa[0] = 0;
LL ans = 0;
for(int i = 1; i <= N; i++) {
int x = *s.begin(); s.erase(s.begin());
int f = find(fa[find(x)]);
if(f) s.erase(f);
ans += siz[f] * w[x];
siz[f] += siz[x]; w[f] += w[x];
ufa[x] = f;
if(f) s.insert(f);
}
cout << ans;
return 0;
}
/*
3
0 1 1
5 7 3
*/
作者:自为风月马前卒
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。