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

Posted 2022-08-06 Herio

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

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

实际上枚举选取的个数。

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

// 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;