E. Madoka and the Sixth-graders

Problem - E - Codeforces

思路：写这个题给我写破防了。。。

首先我们能够发现在一次迭代的过程中，所有入度为0的点都会变成空的，由大于n的几个数去填充，所以我们能够发现在一次迭代的过程中，新增的数的个数是固定的，所以我们可以用最大值减去n表示新增了多少个数，然后除以每次新增的个数，即入度为0的点，我们就能够知道一共进行了多少次迭代，因为每个点的前驱是固定的，所以我们可以知道经过这些次迭代每个点能够到那个位置，这个可以通过倍增来维护，这样就求出来了每个点能够迭代到的位置

举个例子，例如第三个样例1 6 8 6 6 6 8 8 10 10，这是每个点能够迭代到的位置，那么我们能够发现第2 4 5 6，这四个点迭代到了相同的位置6，并且此时6为4，那么我们知道2 4 5 6这四个位置的数肯定都是大于等于4的，为了让字典序最小，我们肯定会把4放到2这个位置上，所以我们可以先将所有小于n的数固定了位置，假如说每个位置能够迭代到的数为dist[i]，那么对于剩下的位置来说，只需要这个位置方的数比w[dist[i]]大即可，同时先满足座位号更小的，再满足更大的，这样能够保证字典序最小化

// Problem: E. Madoka and the Sixth-graders
// Contest: Codeforces - Codeforces Round 777 (Div. 2)
// URL: https://codeforces.com/contest/1647/problem/E
// Memory Limit: 256 MB
// Time Limit: 2000 ms

#include<bits/stdc++.h>
#include<sstream>
#include<cassert>
#define fi first
#define se second
#define i128 __int128
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> PII;
const double eps=1e-7;
const int N=5e5+7 ,M=2e6+7, INF=0x3f3f3f3f,mod=1e9+7,mod1=998244353;
const long long int llINF=0x3f3f3f3f3f3f3f3f;
inline ll read() {ll x=0,f=1;char c=getchar();while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {x=(ll)x*10+c-'0';c=getchar();} return x*f;}
inline void write(ll x) {if(x < 0) {putchar('-'); x = -x;}if(x >= 10) write(x / 10);putchar(x % 10 + '0');}
inline void write(ll x,char ch) {write(x);putchar(ch);}
void stin() {freopen("in_put.txt","r",stdin);freopen("my_out_put.txt","w",stdout);}
bool cmp0(int a,int b) {return a>b;}
template<typename T> T gcd(T a,T b) {return b==0?a:gcd(b,a%b);}
template<typename T> T lcm(T a,T b) {return a*b/gcd(a,b);}
void hack() {printf("\n----------------------------------\n");}

int T,hackT;
int n,m,k;
int p[N];
int w[N];
int dist[N];
int f[N][31];
int d[N];
int ans[N];
bool id[N];

int get(int u,int len) {
	for(int j=30;j>=0;j--) {
		if(len>=(1<<j)) {
			len-=(1<<j);
			u=f[u][j];
		}
	}
	return u;
}

void solve() {
	n=read();
	
	int maxn=-1;
	for(int i=1;i<=n;i++) p[i]=read(),d[p[i]]++;
	for(int i=1;i<=n;i++) w[i]=read(),maxn=max(maxn,w[i]);
	
	int cnt=0;
	for(int i=1;i<=n;i++) if(!d[i]) cnt++;
	
	int len=(maxn-n)/cnt;
	
	for(int i=0;i<=30;i++) {
		if(i==0) {
			for(int j=1;j<=n;j++) f[j][0]=p[j];
			continue;
		}
		for(int j=1;j<=n;j++) f[j][i]=f[f[j][i-1]][i-1];
	}
	
	for(int j=1;j<=n;j++) dist[j]=get(j,len);
	
	set<int> s;
	for(int i=1;i<=n;i++) s.insert(i);
	
	map<int,int> st;
	
	map<int,int> vp;
	for(int i=1;i<=n;i++) {
		int k=dist[i];
		if(w[k]<=n&&vp[k]==0) {
			ans[i]=w[k];
			vp[k]=1;
			st[k]=max(st[k],ans[i]);
			s.erase(s.lower_bound(ans[i]));
		}
	}
	
	for(int i=1;i<=n;i++) {
		if(ans[i]) continue;
		int t=0;
		t=max(t,st[dist[i]]);
		
		auto it=s.lower_bound(t);
		ans[i]=*it;

		s.erase(it);
	}
	
	for(int i=1;i<=n;i++) printf("%d ",ans[i]);
	printf("\n");
}   

int main() {
    // init();
    // stin();
	// ios::sync_with_stdio(false); 

    // scanf("%d",&T);
    T=1; 
    while(T--) hackT++,solve();
    
    return 0;       
}