結果

問題 No.3405 Engineering University of Tree
コンテスト
ユーザー Mike scarf
提出日時 2025-12-12 16:09:47
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 521 ms / 2,000 ms
コード長 2,079 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,564 ms
コンパイル使用メモリ 289,644 KB
実行使用メモリ 64,356 KB
最終ジャッジ日時 2025-12-12 16:10:00
合計ジャッジ時間 12,569 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:47:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   47 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
main.cpp:49:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   49 |         scanf("%d",&c);
      |         ~~~~~^~~~~~~~~
main.cpp:53:18: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   53 |             scanf("%d",&e);
      |             ~~~~~^~~~~~~~~

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=200005;
const ll inf=1e18;
vector<int> s[maxn+5],nd[maxn+5],rn[maxn+5];
vector<int> an;
int dg[maxn+5],pt[maxn+5],vis[maxn+5],ps[maxn+5],vt;
ll f[maxn+5],g[maxn+5];
int n;
bool cmp(int a,int b){return dg[a]<dg[b];}
void dfs(int u,int p,int q){
    vis[u]=vt;
    while(pt[u]<s[u].size()&&dg[s[u][pt[u]]]<q)pt[u]++;
    int ni=pt[u];
    vector<ll> cs;
    ll bs=0,sc=0;
    for(int i=pt[u];i<s[u].size();i++){
        int v=s[u][i];
        if(v==p)continue;
        dfs(v,u,q);
        bs+=f[v];
        if(g[v]!=-inf)cs.push_back(max(0LL,g[v]-f[v]));
        else cs.push_back(0);
    }
    sort(cs.begin(),cs.end());
    for(auto c:cs)sc+=c;
    ll vnc=bs+sc;
    ll vc=-inf;
    if(ni+(int)cs.size()>=q){
        int ndd=max(0,q-ni);
        ll cur=0;
        for(int i=0;i<ndd;i++)cur+=cs[i];
        vc=1+bs+sc-cur;
    }
    f[u]=max(vnc,vc);
    g[u]=-inf;
    if(q>0&&ni+(int)cs.size()>=q-1){
        int ndd=max(0,q-1-ni);
        ll cur=0;
        for(int i=0;i<ndd;i++)cur+=cs[i];
        g[u]=1+bs+sc-cur;
    }
}
int main(){
    int i,j,k,c,u,v,e;
    scanf("%d",&n);
    for(i=1;i<=n;i++){
        scanf("%d",&c);
        dg[i]=c;
        nd[c].push_back(i);
        for(j=0;j<c;j++){
            scanf("%d",&e);
            rn[e].push_back(i);
        }
    }
    for(i=1;i<n;i++){
        u=rn[i][0];v=rn[i][1];
        s[u].push_back(v);
        s[v].push_back(u);
    }
    for(i=1;i<=n;i++){
        sort(s[i].begin(),s[i].end(),cmp);
        an.push_back(i);
        ps[i]=i-1;
    }
    for(i=1;i<n;i++){
        vt++;
        ll ans=0;
        for(auto u:nd[i-1]){
            int idx=ps[u];
            int last=an.back();
            an[idx]=last;
            ps[last]=idx;
            an.pop_back();
        }
        for(auto u:an){
            while(pt[u]<s[u].size()&&dg[s[u][pt[u]]]<i)pt[u]++;
            if(vis[u]!=vt){
                dfs(u,0,i);
                ans+=f[u];
            }
        }
        printf("%lld%c",ans,i==n-1?'\n':' ');
    }
    return 0;
}
0