D - I Hate Non-integer Number(背包dp)

Posted 2022-08-08 Harris-H

D - I Hate Non-integer Number(背包dp)

一开始想到dp，但是开了个三维数组不知道怎么转移。

实际上枚举选取的个数$i$。

然后遍历$a$，枚举当前选了$j$个，模$i$的余数为$k$，这里用背包思想简化一维。

// Problem: D - I Hate Non-integer Number
// Contest: AtCoder - AtCoder Beginner Contest 262
// URL: https://atcoder.jp/contests/abc262/tasks/abc262_d
// Memory Limit: 1024 MB
// Time Limit: 2500 ms
// Date: 2022-07-31 23:51:13
// --------by Herio--------

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=105,M=2e4+5,inf=0x3f3f3f3f,mod=998244353;
const int hashmod[4] = 402653189,805306457,1610612741,998244353;
#define mst(a,b) memset(a,b,sizeof a)
#define db double
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define x first
#define y second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define per(i,a,b) for(int i=a;i>=b;--i)
#define ios ios::sync_with_stdio(false),cin.tie(nullptr) 
void Print(int *a,int n)
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\\n",a[n]); 

template <typename T>		//x=max(x,y)  x=min(x,y)
void cmx(T &x,T y)
	if(x<y) x=y;

template <typename T>
void cmn(T &x,T y)
	if(x>y) x=y;

ll a[N];
ll f[N][N];
ll ans;
int main()
	int n;cin>>n;
	rep(i,1,n) cin>>a[i];
	rep(i,1,n)
		mst(f,0);
		f[0][0]=1;
		for(int u=1;u<=n;u++)
			for(int j=i;j;j--)
				for(int k=0;k<i;k++)
				(f[j][(k+a[u])%i]+=f[j-1][k])%=mod;
			
		(ans+=f[i][0])%=mod;
	
	cout<<ans;
	return 0;