#724915 (C++17) No.1789 Tree Growing

提出ソース

結果

問題	No.1789 Tree Growing
ユーザー	chocorusk
提出日時	2021-12-18 19:28:32
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	330 ms / 5,000 ms
コード長	3,502 bytes
コンパイル時間	5,725 ms
コンパイル使用メモリ	296,092 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-09-15 13:53:26
合計ジャッジ時間	15,760 ms
ジャッジサーバーID （参考情報）	judge1 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	3 ms 5,248 KB
testcase_02	AC	3 ms 5,376 KB
testcase_03	AC	3 ms 5,376 KB
testcase_04	AC	3 ms 5,376 KB
testcase_05	AC	3 ms 5,376 KB
testcase_06	AC	4 ms 5,376 KB
testcase_07	AC	4 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	3 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	3 ms 5,376 KB
testcase_14	AC	3 ms 5,376 KB
testcase_15	AC	6 ms 5,376 KB
testcase_16	AC	3 ms 5,376 KB
testcase_17	AC	10 ms 5,376 KB
testcase_18	AC	8 ms 5,376 KB
testcase_19	AC	5 ms 5,376 KB
testcase_20	AC	10 ms 5,376 KB
testcase_21	AC	52 ms 5,376 KB
testcase_22	AC	29 ms 5,376 KB
testcase_23	AC	38 ms 5,376 KB
testcase_24	AC	25 ms 5,376 KB
testcase_25	AC	84 ms 5,376 KB
testcase_26	AC	68 ms 5,376 KB
testcase_27	AC	66 ms 5,376 KB
testcase_28	AC	89 ms 5,376 KB
testcase_29	AC	52 ms 5,376 KB
testcase_30	AC	114 ms 5,376 KB
testcase_31	AC	122 ms 5,376 KB
testcase_32	AC	95 ms 5,376 KB
testcase_33	AC	83 ms 5,376 KB
testcase_34	AC	129 ms 5,376 KB
testcase_35	AC	90 ms 5,376 KB
testcase_36	AC	100 ms 5,376 KB
testcase_37	AC	148 ms 5,376 KB
testcase_38	AC	88 ms 5,376 KB
testcase_39	AC	90 ms 5,376 KB
testcase_40	AC	111 ms 5,376 KB
testcase_41	AC	99 ms 5,376 KB
testcase_42	AC	98 ms 5,376 KB
testcase_43	AC	133 ms 5,376 KB
testcase_44	AC	128 ms 5,376 KB
testcase_45	AC	128 ms 5,376 KB
testcase_46	AC	124 ms 5,376 KB
testcase_47	AC	140 ms 5,376 KB
testcase_48	AC	121 ms 5,376 KB
testcase_49	AC	151 ms 5,376 KB
testcase_50	AC	153 ms 5,376 KB
testcase_51	AC	159 ms 5,376 KB
testcase_52	AC	106 ms 5,376 KB
testcase_53	AC	129 ms 5,376 KB
testcase_54	AC	325 ms 5,376 KB
testcase_55	AC	254 ms 5,376 KB
testcase_56	AC	327 ms 5,376 KB
testcase_57	AC	201 ms 5,376 KB
testcase_58	AC	330 ms 5,376 KB
testcase_59	AC	175 ms 5,376 KB
testcase_60	AC	136 ms 5,376 KB
testcase_61	AC	95 ms 5,376 KB
testcase_62	AC	174 ms 5,376 KB
testcase_63	AC	194 ms 5,376 KB
testcase_64	AC	69 ms 5,376 KB
testcase_65	AC	122 ms 5,376 KB
testcase_66	AC	43 ms 5,376 KB
testcase_67	AC	63 ms 5,376 KB
testcase_68	AC	87 ms 5,376 KB
testcase_69	AC	59 ms 5,376 KB
testcase_70	AC	70 ms 5,376 KB
testcase_71	AC	58 ms 5,376 KB
testcase_72	AC	318 ms 5,376 KB
testcase_73	AC	226 ms 5,376 KB
testcase_74	AC	156 ms 5,376 KB
testcase_75	AC	2 ms 5,376 KB
testcase_76	AC	2 ms 5,376 KB
testcase_77	AC	11 ms 5,376 KB
testcase_78	AC	12 ms 5,376 KB
testcase_79	AC	8 ms 5,376 KB
testcase_80	AC	170 ms 5,376 KB
testcase_81	AC	160 ms 5,376 KB
testcase_82	AC	145 ms 5,376 KB
testcase_83	AC	93 ms 5,376 KB
testcase_84	AC	95 ms 5,376 KB
testcase_85	AC	92 ms 5,376 KB
testcase_86	AC	99 ms 5,376 KB
testcase_87	AC	91 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
const int MAXN=100;
const int MAXK=100;
const int INF=1e6;
int n, k, vn[MAXN], vk[MAXK], ordn[MAXN], ordk[MAXK];;
vector<int> gn[MAXN], gk[MAXK];
int main()
{
    cin>>k;
	for(int i=0; i<k-1; i++){
        int a, b;
        cin>>a>>b;
        a--; b--;
        gk[a].push_back(b);
        gk[b].push_back(a);
    }
	cin>>n;
    for(int i=0; i<n-1; i++){
        int a, b;
        cin>>a>>b;
        a--; b--;
        gn[a].push_back(b);
        gn[b].push_back(a);
    }
    {
        queue<int> que;
        que.push(0);
        bool used[MAXK]={};
        used[0]=1;
        int t=0;
        while(!que.empty()){
            int x=que.front(); que.pop();
            ordk[x]=t;
            vk[t++]=x;
            for(auto y:gk[x]){
                if(used[y]) continue;
                que.push(y);
                used[y]=1;
            }
        }
    }
    string ts[MAXK];
    int tk[MAXK];
    vector<string> vss(k);
    for(int i=k-1; i>=0; i--){
        int x=vk[i];
        vector<string> vs;
        for(auto y:gk[x]){
            if(ordk[y]<ordk[x]) continue;
            vs.push_back("("+ts[ordk[y]]+")");
        }
        sort(vs.begin(), vs.end());
        ts[i]="";
        for(auto s:vs) ts[i]+=s;
        vss[i]=ts[i];
    }
    sort(vss.begin(), vss.end());
    vss.erase(unique(vss.begin(), vss.end()), vss.end());
    for(int i=0; i<k; i++){
        tk[i]=lower_bound(vss.begin(), vss.end(), ts[i])-vss.begin();
    }

    int ans=-INF;
    for(int r=0; r<n; r++){
        queue<int> que;
        que.push(r);
        bool used[MAXN]={};
        used[r]=1;
        int t=0;
        while(!que.empty()){
            int x=que.front(); que.pop();
            ordn[x]=t;
            vn[t++]=x;
            for(auto y:gn[x]){
                if(used[y]) continue;
                que.push(y);
                used[y]=1;
            }
        }
        int dp[MAXK][MAXN];
        bool ok[MAXK][MAXN]={};
        for(int j=0; j<vss.size(); j++) for(int i=0; i<n; i++) dp[j][i]=-INF;

        for(int i=n-1; i>=0; i--){
            for(int j=k-1; j>=0; j--){
                if(!(i==0 && j==0) && ok[tk[j]][i]) continue;
                int x=vn[i], y=vk[j], j1=tk[j];
                ok[j1][i]=1;
                for(auto z:gn[x]){
                    if(ordn[z]<ordn[x]) continue;
                    dp[j1][i]=max(dp[j1][i], dp[j1][ordn[z]]+1);
                }
                int n1=gn[vn[i]].size(), k1=gk[vk[j]].size();
                if(i>0) n1--;
                if(j>0) k1--;
                if(k1>n1) continue;
				mcf_graph<int, int> g(k1+n1+2);
				int s=k1+n1, t=s+1;
				for(int l=0; l<k1; l++) g.add_edge(s, l, 1, 0);
				for(int l=0; l<n1; l++) g.add_edge(l+k1, t, 1, 0);
                int i1=0;
                for(auto z:gn[x]){
                    if(ordn[z]<ordn[x]) continue;
                    int j1=0;
                    for(auto w:gk[y]){
                        if(ordk[w]<ordk[y]) continue;
						g.add_edge(j1, i1+k1, 1, INF-dp[tk[ordk[w]]][ordn[z]]);
						j1++;
                    }
                    i1++;
                }
                auto res=g.flow(s, t);
				if(res.first==k1){
					if(i==0 && j==0) ans=max(ans, INF*k1-res.second);
					dp[j1][i]=max(dp[j1][i], INF*k1-res.second);
                }
            }
        }
    }
	if(ans<0){
		cout<<-1<<endl;
		return 0;
	}
	ans+=k-1;
    cout<<ans<<endl;
    return 0;
}

yukicoder

結果

テストケース

ソースコード