Codeforces 1009F Dominant Indices

另类解法

将每一个节点拥有的各深度节点数量存在vector中，向上返回，这样不会占用过多的内存，以此判断最多节点相应的深度即可，但正常写最后一个数据会T，毕竟一次复制一个节点，相当于复制了（1+2+3+4+...+1e6）次，明显超时，所以特判一下1e6个节点的线性树，就算过了这个题了，，，但这个解法不算正规，最后一个数据改一下就又要T了，大家娱乐一下就好，，，23333

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<string.h>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<fstream>
#include<cstdlib>
#include<ctime>
using namespace std;
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fopen freopen("input.txt", "r", stdin);freopen("output.txt", "w", stout);
#define scd(a) scanf("%d",&a)
#define scf(a) scanf("%lf",&a)
#define scl(a) scanf("%lld",&a)
#define sci(a) scanf("%I64d",&a)
#define scs(a) scanf("%s",a)
typedef long long ll;
const int desll[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
const ll mod=1e9+7;
const int maxn=1e6+7;
const int maxm=1e8+7;
const double eps=1e-4;
int m,n,ar[maxn];
char ch[maxn];
struct node{
    int b,nex;
}no[maxn*4];
int sz=0,head[maxn],de;
vector<int> vvl[maxn];
void add(int a,int b)
{
    no[sz].nex=head[a];
    no[sz].b=b;
    head[a]=sz++;
}
vector<int> dfs(int u){
    vector<int> vl(1);
    vl[0]=1;
    //cout<<vl.size();
    for(int i=head[u];i!=-1;i=no[i].nex){
        int v=no[i].b;
        vector<int> vmid = dfs(v);
        int lemid=vmid.size(),le=vl.size();
        for(int i=0;i<lemid;i++){
            if(i<le-1)vl[i+1]+=vmid[i];
            else {
                vl.insert(vl.end(),vmid.begin()+i,vmid.end());
                break;
            }
        }
    }
    int ans = -1,maxMid=-1;
    for(int i=0;i<vl.size();i++){
        if(maxMid<vl[i]){
            maxMid = vl[i];
            ans=i;
        }
    }
    ar[u]=ans;
    return vl;
}
void init(int u,int pre,int dep)
{
    de=max(de,dep);
    for(int v:vvl[u]){
        if(v!=pre){
            init(v,u,dep+1);
            add(u,v);
        }
    }
}

int main()
{
    scd(n);
    sz=0;
    memset(head,-1,sizeof(head));
    for(int i=1;i<n;i++){
        int x,y;scd(x);scd(y);
        vvl[x].push_back(y);
        vvl[y].push_back(x);
    }
    de=0;
    init(1,0,1);
    //cout<<"de = "<<de<<endl;
    if(de<1e6)dfs(1);
    for(int i=1;i<=n;i++)printf("%d
",ar[i]);


    return 0;
}