板子题...并没有太多的变化.和之前做的,唯一的变化就是对于字符串可以哈希成数字的变成点权.
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+50;
//可以用map把每个字符串做一次离散化,这样就相当于查询权值了.
map<string,int>mp;
vector<int>v[N<<1];
int w[N];//存这个点的权值.
int dfn[N],Son,sz[N];
int dep[N];
set<int>s[N<<1];
struct Ans{
int k,id,res;
};
vector<Ans>cnt[N];
int ans[N];
void dfs(int u,int fa)
{
sz[u]=1;dep[u]+=dep[fa]+1;
for(int i=0;i<(int)v[u].size();i++)
{
int son=v[u][i];
if(son==fa) continue;
dfs(son,u);sz[u]+=sz[son];
if(sz[son]>sz[dfn[u]]) dfn[u]=son;
}
}
void cal(int u,int fa,bool op)
{
if(op) s[dep[u]].insert(w[u]);
else s[dep[u]].clear();
for(int i=0;i<(int)v[u].size();i++)
{
int son=v[u][i];
if(son==fa||son==Son) continue;
cal(son,u,op);
}
}
void dfs(int u,int fa,bool op)
{
for(int i=0;i<(int)v[u].size();i++)
{
int son=v[u][i];
if(son==dfn[u]||son==fa) continue;
dfs(son,u,true);
}
if(dfn[u]) dfs(dfn[u],u,false),Son=dfn[u];
cal(u,fa,true);
for(int i=0;i<(int)cnt[u].size();i++)
{
int Dep=dep[u]+cnt[u][i].k;
cnt[u][i].res=s[Dep].size();
}
Son=0;
if(op)
{
cal(u,fa,false);
}
}
int main()
{
int n,mk=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
string a;int u;
cin>>a>>u;
if(!mp[a]) mp[a]=++mk;
w[i]=mp[a];
v[i].push_back(u);
v[u].push_back(i);
}//处理权值.
dfs(0,0);
int m;
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
int u,Dep;
scanf("%d%d",&u,&Dep);
cnt[u].push_back({Dep,i,0});
}
dfs(0,0,true);
for(int i=1;i<=n;i++)
{
for(int j=0;j<(int)cnt[i].size();j++)
{
ans[cnt[i][j].id]=cnt[i][j].res;
}
}
for(int i=1;i<=m;i++)
{
printf("%d\n",ans[i]);
}
return 0;
}
还有一题就不写专心写题了,学线代汇编物理马原之类的去了,随便去上课~

京公网安备 11010502036488号