标签：std rt 17 int dfs dep 2022.7 mx 模拟

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;
}

标签：std,rt,17,int,dfs,dep,2022.7,mx,模拟
来源： https://www.cnblogs.com/into-qwq/p/16486746.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。