結果

問題 No.317 辺の追加
コンテスト
ユーザー anta
提出日時 2015-12-10 01:18:01
言語 C++11（廃止可能性あり）
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 87 ms / 2,000 ms
コード長 2,184 bytes
記録
コンパイル時間 705 ms
コンパイル使用メモリ 44,672 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-15 07:14:59
合計ジャッジ時間 4,623 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 38
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:59:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   59 |                         scanf("%d%d", &u, &v), -- u, -- v;
      |                         ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff #
raw source code 

#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);
	if(n % 2 == 0) {
		int &x = dp[n + s], y = dp[n] + t;
		if(x > y) x = y;
		if(-- n < 0) return;
	}
#ifndef __GNUC__
	const __m128i tttt = _mm_set1_epi32(t);
	for(int i = n - 1; i >= 0; i -= 2) {
		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 - 1; i >= 0; i -= 2) {
		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;
}
0