結果

問題 No.1530 Permutation and Popcount
コンテスト
ユーザー kotatsugame
提出日時 2021-06-04 22:08:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 944 bytes
コンパイル時間 739 ms
コンパイル使用メモリ 74,672 KB
実行使用メモリ 6,844 KB
最終ジャッジ日時 2024-11-24 20:37:31
合計ジャッジ時間 2,514 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:14:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   14 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<bitset>
#include<vector>
using namespace std;
int N,M;
int cnt[301];
bitset<90001>dp[301];
int f(vector<int>A)
{
	int ret=0;
	for(int a:A)for(int b:A)ret+=!__builtin_parity(a*b);
	return ret;
}
main()
{
	cin>>N>>M;
	for(int i=1;i<=N;i++)
	{
		for(int j=1;j<=N;j++)
		{
			int t=!__builtin_parity(i*j);
			cnt[i]+=t;
		}
		M-=cnt[i];
		//cnt[i]+=!__builtin_parity(i*i);
		cnt[i]*=2;
	}
	M=-M;
	if(M<0)
	{
		cout<<-1<<endl;
		return 0;
	}
	dp[0][0]=1;
	for(int i=1;i<=N;i++)
	{
		dp[i]=dp[i-1]|dp[i-1]<<cnt[i];
	}
	if(dp[N][M]==0)cout<<-1<<endl;
	else
	{
		vector<int>X,Y;
		for(int i=N;i>0;i--)
		{
			if(M>=cnt[i]&&dp[i-1][M-cnt[i]])
			{
				Y.push_back(i);
				M-=cnt[i];
			}
			else X.push_back(i);
		}
		cout<<X.size()<<" "<<Y.size()<<endl;
		for(int i=0;i<X.size();i++)cout<<X[i]<<(i+1==X.size()?"\n":" ");
		for(int i=0;i<Y.size();i++)cout<<Y[i]<<(i+1==Y.size()?"\n":" ");
		cerr<<f(X)<<" "<<f(Y)<<endl;
	}
}
0