#64376 (C++11) No.317 辺の追加

• 問題: No.317 辺の追加
• ユーザー: anta
• 提出日時: 2015-12-10 01:22:01
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 51 ms / 2,000 ms
• コード長: 2,180 bytes
• コンパイル時間: 539 ms
• コンパイル使用メモリ: 44,608 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-15 07:15:21
• 合計ジャッジ時間: 3,838 ms
• ジャッジサーバーID （参考情報）: judge4 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	30 ms; 5,376 KB
testcase_03	AC	35 ms; 5,376 KB
testcase_04	AC	25 ms; 5,376 KB
testcase_05	AC	37 ms; 5,376 KB
testcase_06	AC	44 ms; 5,376 KB
testcase_07	AC	16 ms; 5,376 KB
testcase_08	AC	27 ms; 5,376 KB
testcase_09	AC	36 ms; 5,376 KB
testcase_10	AC	50 ms; 5,376 KB
testcase_11	AC	15 ms; 5,376 KB
testcase_12	AC	49 ms; 5,376 KB
testcase_13	AC	18 ms; 5,376 KB
testcase_14	AC	41 ms; 5,376 KB
testcase_15	AC	47 ms; 5,376 KB
testcase_16	AC	30 ms; 5,376 KB
testcase_17	AC	43 ms; 5,376 KB
testcase_18	AC	50 ms; 5,376 KB
testcase_19	AC	51 ms; 5,376 KB
testcase_20	AC	29 ms; 5,376 KB
testcase_21	AC	16 ms; 5,376 KB
testcase_22	AC	9 ms; 5,376 KB
testcase_23	AC	9 ms; 5,376 KB
testcase_24	AC	27 ms; 5,376 KB
testcase_25	AC	20 ms; 5,376 KB
testcase_26	AC	21 ms; 5,376 KB
testcase_27	AC	17 ms; 5,376 KB
testcase_28	AC	10 ms; 5,376 KB
testcase_29	AC	3 ms; 5,376 KB
testcase_30	AC	32 ms; 5,376 KB
testcase_31	AC	32 ms; 5,376 KB
testcase_32	AC	32 ms; 5,376 KB
testcase_33	AC	32 ms; 5,376 KB
testcase_34	AC	32 ms; 5,376 KB
testcase_35	AC	32 ms; 5,376 KB
testcase_36	AC	32 ms; 5,376 KB
testcase_37	AC	32 ms; 5,376 KB
testcase_38	AC	32 ms; 5,376 KB
testcase_39	AC	32 ms; 5,376 KB

コンパイルメッセージ

main.cpp: In function ‘int main()’:
main.cpp:58:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   58 |                         scanf("%d%d", &u, &v), -- u, -- v;
      |                         ~~~~~^~~~~~~~~~~~~~~~

ソースコード

#include <cstdio>
#include <vector>
#include <algorithm>
#include <cassert>
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#ifndef __GNUC__
#include <intrin.h>
#endif
using namespace std;

struct UnionFind {
	vector<int> data;
	void init(int n) { data.assign(n, -1); }
	bool unionSet(int x, int y) {
		x = root(x); y = root(y);
		if(x != y) {
			if(data[y] < data[x]) swap(x, y);
			data[x] += data[y]; data[y] = x;
		}
		return x != y;
	}
	bool findSet(int x, int y) { return root(x) == root(y); }
	int root(int x) { return data[x] < 0 ? x : data[x] = root(data[x]); }
	int size(int x) { return -data[root(x)]; }
};

void update(int *dp, int n, int s, int t) {
	assert(s > 1);
	for(; n >= 0 && n % 4 != 3; -- n) {
		int &x = dp[n + s], y = dp[n] + t;
		if(x > y) x = y;
	}
#ifndef __GNUC__
	const __m128i tttt = _mm_set1_epi32(t);
	for(int i = n - 3; i >= 0; i -= 4) {
		const __m128i x = _mm_loadu_si128(reinterpret_cast<const __m128i*>(dp + (i + s)));
		const __m128i y = _mm_add_epi32(_mm_loadu_si128(reinterpret_cast<const __m128i*>(dp + i)), tttt);
		const __m128i z = _mm_min_epi32(x, y);
		_mm_storeu_si128(reinterpret_cast<__m128i*>(dp + (i + s)), z);
	}
#else
	__asm("vmovd %0, %%xmm2\n\tvpshufd $0, %%xmm2, %%xmm1":: "r"(t) : "xmm1", "xmm2");
	for(int i = n - 3; i >= 0; i -= 4) {
		const int * const p = dp + i;
		int * const q = dp + (i + s);
		__asm("vpaddd %1, %%xmm1, %%xmm0\n\tvpminsd %0, %%xmm0, %%xmm0\n\tvmovups %%xmm0, %0"
			: "=m"(*q) : "m"(*p) : "xmm0", "xmm1");
	}
#endif
}

int main() {
	int N; int M;
	while(~scanf("%d%d", &N, &M)) {
		UnionFind uf; uf.init(N);
		rep(i, M) {
			int u; int v;
			scanf("%d%d", &u, &v), -- u, -- v;
			uf.unionSet(u, v);
		}
		vector<int> counts(N + 1);
		rep(i, N) if(uf.root(i) == i)
			++ counts[uf.size(i)];
		const int INF = 0x3f3f3f3f;
		vector<int> dp(N + 1, INF);
		for(int n = counts[1]; n >= 0; -- n)
			dp[n] = n;
		int sum = counts[1];
		for(int i = 2; i <= N; ++ i) {
			int n = counts[i];
			while(n > 0) {
				int m = (n + 1) / 2;
				update(&dp[0], sum, i * m, m);
				sum += i * m;
				n -= m;
			}
		}
		for(int i = 1; i <= N; ++ i)
			printf("%d\n", dp[i] == INF ? -1 : dp[i] - 1);
	}
	return 0;
}