2022.7.17 模拟赛

2022.7.17 模拟赛#

$\to link \leftarrow$

订货#

简单题，但为什么 $1\le n\le 2\times 10^6$ 的数据我 $n\log n$ 做法会 $T$ ？

我不理解

模糊匹配#

模拟题

国境线#

$std$ 貌似锅了，直接把 $std$ 交上去了

来自某同学的一句话：$\color{Green}{\Huge{我跟\ std\ 锅都锅的一样}}$

小树精灵#

人类智慧题

很显然要求的三个点交于一点

我们枚举这个交点，以这个交点为根选 $3$ 个深度不同的点即为一组答案

#include<bits/stdc++.h>
using namespace std;

#define int long long

const int N=3005;

int n,ans,mx;
vector <int> G[N];
int f[N][4],cnt[N][N];

inline void dfs(int x,int fa,int rt,int dep){
	++cnt[rt][dep];
	mx=max(mx,dep);
	for(auto y:G[x])
		if(y!=fa) dfs(y,x,rt,dep+1);
}

signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;++i){
		int x;
		cin>>x;
		if(x) G[x].push_back(i),G[i].push_back(x);
	}
	for(int i=1;i<=n;++i){
		for(int j=1;j<=n;++j)
			f[j][0]=1,f[j][1]=f[j][2]=f[j][3]=0;
		for(auto x:G[i]){
			for(int j=1;j<=n;++j) cnt[x][j]=0;
			mx=0;
			dfs(x,i,x,1);
			for(int j=1;j<=mx;++j)
				for(int k=3;k;--k)
					f[j][k]+=f[j][k-1]*cnt[x][j];
		}
		for(int j=1;j<=n;++j)
			ans+=f[j][3];
	}
	cout<<ans<<endl;
}