#663553 (C) No.1530 Permutation and Popcount

• 問題: No.1530 Permutation and Popcount
• ユーザー: tails
• 提出日時: 2021-06-04 21:18:54
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 44 ms / 2,000 ms
• コード長: 1,382 bytes
• コンパイル時間: 1,151 ms
• コンパイル使用メモリ: 32,768 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-03 17:33:07
• 合計ジャッジ時間: 2,333 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	0 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	1 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	AC	1 ms; 5,376 KB
testcase_15	AC	1 ms; 5,376 KB
testcase_16	AC	1 ms; 5,376 KB
testcase_17	AC	22 ms; 5,376 KB
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	1 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	1 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	1 ms; 5,376 KB
testcase_29	AC	1 ms; 5,376 KB
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	1 ms; 5,376 KB
testcase_32	AC	44 ms; 5,376 KB
testcase_33	AC	1 ms; 5,376 KB
testcase_34	AC	1 ms; 5,376 KB
testcase_35	AC	44 ms; 5,376 KB
testcase_36	AC	1 ms; 5,376 KB
testcase_37	AC	1 ms; 5,376 KB
testcase_38	AC	1 ms; 5,376 KB
testcase_39	AC	44 ms; 5,376 KB
testcase_40	AC	43 ms; 5,376 KB
testcase_41	AC	1 ms; 5,376 KB
testcase_42	AC	43 ms; 5,376 KB
testcase_43	AC	17 ms; 5,376 KB

コンパイルメッセージ

main.c:21:1: warning: return type defaults to 'int' [-Wimplicit-int]
   21 | main(){
      | ^~~~
main.c: In function 'main':
main.c:23:9: warning: implicit declaration of function 'read'; did you mean 'rd'? [-Wimplicit-function-declaration]
   23 |         read(0,rbuf,sizeof rbuf);
      |         ^~~~
      |         rd
main.c:37:17: warning: implicit declaration of function 'write' [-Wimplicit-function-declaration]
   37 |                 write(1,"-1\n",3);
      |                 ^~~~~
main.c:38:17: warning: implicit declaration of function '_exit'; did you mean '_Exit'? [-Wimplicit-function-declaration]
   38 |                 _exit(0);
      |                 ^~~~~
      |                 _Exit

ソースコード

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

#define rd_skip() while(*rp++>=48)
#define rd(v) int v=0;{int _c;while(_c=*rp++-48,_c>=0)v=v*10+_c;}

char wbuf[1<<25];

#define WTHI(v) {long _z=v,_n=0,_d=0;while(++_n,_d=_d<<8|0x30|_z%10,_z/=10);*(long*)wp=_d;wp+=_n;}
#define WTLO(v) {long _z=v,_n=8,_d=0;while(_d=_d<<8|0x30|_z%10,_z/=10,--_n);*(long*)wp=_d;wp+=8;}
#define wt(v) WTHI(v)

int count[301];
int total;

typedef struct {
	int a[10];
} Bits; 
Bits dp[200000];

main(){
	char rbuf[64];
	read(0,rbuf,sizeof rbuf);
	char*rp=rbuf;
	rd(n);
	rd(m);

	for(int i=1;i<=n;++i){
		for(int j=1;j<=n;++j){
			int p=!__builtin_parity(i*j);
			count[i]+=p;
			total+=p;
		}
	}

	if((total^m)&1 || m>total){
		write(1,"-1\n",3);
		_exit(0);
	}
	m=total-m>>1;

	for(int k=1;k<=m;++k){
		dp[k].a[9]=-1;
	}

	for(int i=1;i<=n;++i){
		for(int k=m;k>=count[i];--k){
			Bits b=dp[k-count[i]];
			if(b.a[9]>=0){
				b.a[i>>5]|=1<<i;
				dp[k]=b;
			}
		}
	}

	if(dp[m].a[9]<0){
		write(1,"-1\n",3);
		_exit(0);
	}

	int yn=0;
	for(int j=0;j<10;++j){
		yn+=__builtin_popcount(dp[m].a[j]);
	}

	char*wp=wbuf;
	wt(n-yn);
	*wp++=' ';
	wt(yn);
	*wp++='\n';
	for(int i=1;i<=n;++i){
		if(!(dp[m].a[i>>5]&1<<i)){
			wt(i);
			*wp++=' ';
		}
	}
	wp[-1]='\n';
	for(int i=1;i<=n;++i){
		if(dp[m].a[i>>5]&1<<i){
			wt(i);
			*wp++=' ';
		}
	}
	wp[-1]='\n';

	write(1,wbuf,wp-wbuf);
	_exit(0);
}