【计数DP】ABC230 F - Predilection

最新推荐文章于 2024-06-09 20:18:37 发布

lamentropetion

最新推荐文章于 2024-06-09 20:18:37 发布

阅读量47

点赞数

分类专栏：动态规划组合数学文章标签： c++ 动态规划算法

本文链接：https://blog.csdn.net/weixin_62528401/article/details/130618971

版权

动态规划同时被 2 个专栏收录

144 篇文章 1 订阅

订阅专栏

组合数学

37 篇文章 0 订阅

订阅专栏

计数DP，会不了一点....

F - Predilection (atcoder.jp)

思路：

Code：

#include <bits/stdc++.h>

#define int long long
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)

using namespace std;

const int mxn=2e5+10;
const int mxe=2e5+10;
const int mod=998244353;

map<int,int> mp;

int N,sum=0;
int a[mxn],dp[mxn];

void solve(){
	cin>>N;
	for(int i=1;i<=N;i++) cin>>a[i];
	int res=1ll;
	for(int i=1;i<=N;i++){
		sum+=a[i];
		dp[i]=res;
		res=(res*2ll)%mod;
		res=((res-dp[mp[sum]])%mod+mod)%mod;
		mp[sum]=i;
	}
	cout<<dp[N]<<'\n';
}
signed main(){
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int __=1;//cin>>__;
    while(__--)solve();return 0;
}