結果

問題 No.1634 Sorting Integers (Multiple of K) Hard
ユーザー publfl
提出日時 2021-07-30 21:29:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,524 ms / 3,000 ms
コード長 1,472 bytes
コンパイル時間 871 ms
コンパイル使用メモリ 71,480 KB
実行使用メモリ 88,832 KB
最終ジャッジ日時 2024-09-15 23:07:04
合計ジャッジ時間 47,815 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <vector>
#include <algorithm>
#include <map>

int limit,a,b;
std::vector<int> V,V1,V2;
std::map<long long int,int> M,check1,check2;

long long int P[110];
long long int ans = 0;
void func(int k)
{
	if(k>=a)
	{
		if(V1.size()!=limit) return;
		
		long long int temp = 0;
		for(int i=0;i<limit;i++) temp += P[i]*V1[i];
		if(check1[temp]>0) return;
		
		M.clear();
		check2.clear();
		std::vector<int> W1 = V1, W2 = V2;
		std::sort(W1.begin(),W1.end());
		std::sort(W2.begin(),W2.end());
		do{
			long long int sum = 0;
			for(int i=0;i<limit;i++) sum += P[i]*W1[i];
			if(check1[sum]>0) continue;
			check1[sum] = 1;
			M[sum%b]++;
		}while(std::next_permutation(W1.begin(),W1.end()));
		
		do{
			long long int sum = 0;
			for(int i=limit;i<a;i++) sum += P[i]*W2[i-limit];
			if(check2[sum]>0) continue;
			
			check2[sum] = 1;
			ans += M[(b-sum%b)%b];
		}while(std::next_permutation(W2.begin(),W2.end()));
		return;
	}
	
	V1.push_back(V[k]);
	func(k+1);
	V1.pop_back();
	
	V2.push_back(V[k]);
	func(k+1);
	V2.pop_back();
	
}

int c[110];

int main()
{
	scanf("%d%d",&a,&b);
	for(int i=1;i<=9;i++) scanf("%d",&c[i]);
	if(a==1)
	{
		for(int i=1;i<=9;i++)
		{
			if(c[i]==1)
			{
				if(i%b==0) printf("1");
				else printf("0");
				return 0;
			}
		}
	}
	
	for(int i=1;i<=9;i++) while(c[i]--) V.push_back(i);
	P[0] = 1;
	for(int i=1;i<a;i++) P[i] = (10*P[i-1]);
	if(a%2==0) limit = a/2;
	else limit = (a+1)/2;
	
	func(0);
	printf("%lld",ans);
}
0