結果

問題 No.1912 Get together 2
ユーザー 👑 ygussanyygussany
提出日時 2022-04-22 23:25:47
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 163 ms / 2,000 ms
コード長 975 bytes
コンパイル時間 577 ms
コンパイル使用メモリ 31,104 KB
実行使用メモリ 16,636 KB
最終ジャッジ日時 2024-06-24 04:52:59
合計ジャッジ時間 4,955 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>

const int bit[21] = {1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576};

void chmax(long long* a, long long b)
{
	if (*a < b) *a = b;
}

int main()
{
	int i, N, M, V[100001];
	char S[100001][21];
	scanf("%d %d", &N, &M);
	for (i = 1; i <= N; i++) scanf("%d", &(V[i]));
	for (i = 1; i <= N; i++) scanf("%s", S[i]);
	
	int j, k, l, sum[1048576] = {};
	for (i = 1; i <= N; i++) {
		for (j = 0, k = 0; j < M; j++) if (S[i][j] == 'x') k |= bit[j];
		sum[k] += V[i];
	}
	for (j = 0; j < M; j++) {
		for (k = bit[M] - 1, l = bit[M] - 1 - bit[j]; k >= 0; k--) {
			k &= l;
			sum[k] += sum[k ^ bit[j]];
		}
	}
	
	long long dp[1048576] = {}, tmp;
	for (k = 0; k < bit[M]; k++) {
		for (j = 0; j < M; j++) {
			if ((k & bit[j]) != 0) continue;
			tmp = sum[k ^ bit[j]] - sum[k];
			chmax(&(dp[k ^ bit[j]]), dp[k] + tmp * tmp);
		}
	}
	printf("%lld\n", dp[bit[M] - 1]);
	fflush(stdout);
	return 0;
}
0