結果

問題 No.3079 Unite Japanese Prefectures
ユーザー askr58
提出日時 2025-03-28 23:16:21
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,971 bytes
コンパイル時間 6,688 ms
コンパイル使用メモリ 337,360 KB
実行使用メモリ 7,324 KB
最終ジャッジ日時 2025-03-28 23:16:35
合計ジャッジ時間 12,099 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 5 WA * 1 TLE * 1 -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;

int main()
{
	ll n,m;
	cin>>n>>m;
	vector<vector<pair<int,int>>> edges(6);
	for(int i=0;i<m;i++){
		int u,v,c;
		cin>>u>>v>>c;
		u--;v--;c--;
		edges[c].push_back(make_pair(u,v));
	}
	unordered_set<ll> ok_set;
	for(int a=0;a<n;a++){
		for(int b=0;a+b<n;b++){
			for(int c=0;a+b+c<n;c++){
				for(int d=0;a+b+c+d<n;d++){
					for(int e=0;a+b+c+d+e<n;e++){
						int f=n-1-(a+b+c+d+e);
						dsu uf(n);
						int cnt=0;
						for(auto[u,v]:edges[5]){
							if(cnt==a)break;
							if(!uf.same(u,v)){
								cnt++;
								uf.merge(u,v);
							}
						}for(auto[u,v]:edges[4]){
							if(cnt==a+b)break;
							if(!uf.same(u,v)){
								cnt++;
								uf.merge(u,v);
							}
						}for(auto[u,v]:edges[3]){
							if(cnt==a+b+c)break;
							if(!uf.same(u,v)){
								cnt++;
								uf.merge(u,v);
							}
						}for(auto[u,v]:edges[2]){
							if(cnt==a+b+c+d)break;
							if(!uf.same(u,v)){
								cnt++;
								uf.merge(u,v);
							}
						}for(auto[u,v]:edges[1]){
							if(cnt==a+b+c+d+e)break;
							if(!uf.same(u,v)){
								cnt++;
								uf.merge(u,v);
							}
						}for(auto[u,v]:edges[0]){
							if(cnt==n-1)break;
							if(!uf.same(u,v)){
								cnt++;
								uf.merge(u,v);
							}
						}
						if(cnt==n-1)ok_set.insert(a+n*b+n*n*c+n*n*n*d+n*n*n*n*e+n*n*n*n*n*f);
					}
				}
			}
		}
	}

	unordered_map<ll,double> dp;
	auto rec=[&](auto rec,ll s)->double{
		if(dp.count(s))return dp[s];
		ll v[6];
		ll t=s;
		for(int i=0;i<6;i++){
			v[i]=t%n;
			t/=n;
		}
		if(v[0]+v[1]+v[2]+v[3]+v[4]+v[5]==n-1){
			if(ok_set.count(s))return 0;
			else return -1;
		}
		double res=0;
		int cnt=0;
		t=1;
		for(int i=0;i<6;i++){
			double r=rec(rec,s+t);
			if(r!=-1){
				cnt++;
				res+=r;
			}
			t*=n;
		}
		if(cnt==0)return dp[s]=-1;
		else return dp[s]=(6+res)/cnt;
	};

	cout<<fixed<<setprecision(10);
	cout<<rec(rec,0)<<endl;
}
0