#293145 (C++14) No.748 yuki国のお財布事情

• 問題: No.748 yuki国のお財布事情
• ユーザー: kotatsugame
• 提出日時: 2018-10-19 23:32:33
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 139 ms / 2,000 ms
• コード長: 1,036 bytes
• コンパイル時間: 1,170 ms
• コンパイル使用メモリ: 81,360 KB
• 実行使用メモリ: 6,620 KB
• 最終ジャッジ日時: 2024-11-18 22:44:52
• 合計ジャッジ時間: 3,045 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	1 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	2 ms; 5,248 KB
testcase_12	AC	2 ms; 5,248 KB
testcase_13	AC	16 ms; 5,248 KB
testcase_14	AC	28 ms; 5,248 KB
testcase_15	AC	20 ms; 5,248 KB
testcase_16	AC	52 ms; 5,248 KB
testcase_17	AC	100 ms; 5,852 KB
testcase_18	AC	121 ms; 5,976 KB
testcase_19	AC	139 ms; 6,492 KB
testcase_20	AC	122 ms; 6,108 KB
testcase_21	AC	116 ms; 6,108 KB
testcase_22	AC	2 ms; 5,248 KB
testcase_23	AC	2 ms; 5,248 KB
testcase_24	AC	2 ms; 5,248 KB
testcase_25	AC	96 ms; 5,952 KB
testcase_26	AC	115 ms; 6,620 KB
testcase_27	AC	111 ms; 6,368 KB
testcase_28	AC	94 ms; 5,728 KB

コンパイルメッセージ

main.cpp:40:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   40 | main()
      | ^~~~

ソースコード

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int n,m,k;
int a[1<<17],b[1<<17],c[1<<17];
long sum;
#include<vector>
struct UF{
	int n;
	vector<int>parent,rank;
	UF(int n_=0):n(n_),parent(n_),rank(n_,0)
	{
		for(int i=0;i<n_;i++)parent[i]=i;
	}
	int find(int a)
	{
		return parent[a]!=a?parent[a]=find(parent[a]):a;
	}
	bool same(int a,int b)
	{
		return find(a)==find(b);
	}
	bool unite(int a,int b)
	{
		a=find(a),b=find(b);
		if(a==b)return false;
		if(rank[a]<rank[b])
		{
			parent[a]=b;
		}
		else
		{
			parent[b]=a;
			if(rank[a]==rank[b])rank[a]++;
		}
		return true;
	}
};
main()
{
	cin>>n>>m>>k;
	vector<pair<int,pair<int,int> > >G;
	for(int i=0;i<m;i++)
	{
		cin>>a[i]>>b[i]>>c[i];
		sum+=c[i];
		G.push_back(make_pair(c[i],make_pair(a[i],b[i])));
	}
	UF uf(n+1);
	for(int i=0;i<k;i++)
	{
		int e;cin>>e;
		e--;
		sum-=c[e];
		uf.unite(a[e],b[e]);
	}
	sort(G.begin(),G.end());
	for(int i=0;i<m;i++)
	{
		if(uf.unite(G[i].second.first,G[i].second.second))sum-=G[i].first;
	}
	cout<<sum<<endl;
}