結果

問題 No.1621 Sequence Inversions
ユーザー publflpublfl
提出日時 2021-07-22 22:43:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 172 ms / 3,000 ms
コード長 1,508 bytes
コンパイル時間 863 ms
コンパイル使用メモリ 68,024 KB
実行使用メモリ 221,952 KB
最終ジャッジ日時 2024-07-17 19:05:10
合計ジャッジ時間 3,910 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <algorithm>
#include <vector>
#include <map>
#define MOD 998244353

long long int check[110][10010], check2[110][2510][110];
long long int func2(int k, int s, int t)
{
	if(s<0) return 0;
	if(t<0) return 0;
	if(k==0)
	{
		if(s==0) return 1;
		else return 0;
	}
	if(check2[k][s][t]!=-1) return check2[k][s][t];
	
	long long int s1 = func2(k,s,t-1);
	long long int s2 = func2(k-1,s-t,t);
	return check2[k][s][t] = (s1+s2)%MOD;
}

int b,c;
int sum[110],count[110];
long long int func(int k, int C)
{
	if(C>b) return 0;
	if(k>c)
	{
		if(C==b) return 1;
		else return 0;
	}
	if(check[k][C]!=-1) return check[k][C];
	
	int N = sum[k-1];
	//0 ~ N
	
	long long int ans = 0;
	for(int i=0;i<=count[k]*N;i++)
	{
		long long int t = func(k+1,C+i) * func2(count[k],i,N);
		t %= MOD, ans += t, ans %= MOD;
	}
	return check[k][C] = ans;
}

std::vector<int> index;
std::map<int,int> hash;
int x[110];
int main()
{
	int a;
	scanf("%d%d",&a,&b);
	for(int i=1;i<=a;i++) scanf("%d",&x[i]);
	for(int i=1;i<=a;i++) index.push_back(x[i]);
	std::sort(index.begin(),index.end());
	index.erase(std::unique(index.begin(),index.end()),index.end());
	for(int i=0;i<index.size();i++) hash[index[i]] = i+1;
	
	c = index.size();
	for(int i=1;i<=a;i++) count[hash[x[i]]]++;
	for(int i=1;i<=c;i++) sum[i] = sum[i-1] + count[i];
	
	for(int i=1;i<=c;i++) for(int j=0;j<=b;j++) check[i][j] = -1;
	for(int i=1;i<=a;i++) for(int j=0;j<=2500;j++) for(int k=0;k<=a;k++) check2[i][j][k] = -1;
	
	printf("%lld",func(1,0));
}
0