UKE求調
查看原帖
UKE求調
653100
_Silica_楼主2023/7/17 13:50
#include <bits/stdc++.h>
using namespace std;
const int N=1e6+5;
int n,a[N],dp[N][2],ans=0;
char s[N];
int main()
{
	scanf("%d\n",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&s[i]);
	}
	for(int i=1;i<=n;i++)
	{
	    a[i]=s[i-1]-'0';
	}
	if(a[1]==0)
	{
	    dp[1][0]=1;
	}
	else
	{
		dp[1][1]=1;
	}
	for(int i=2;i<=n;i++)
	{
		if(a[i]==0)
		{
			dp[i][1]=dp[i-1][0]+dp[i-1][1];
			dp[i][0]=1;
		}
		else
		{
			dp[i][1]=dp[i-1][0]+1;
			dp[i][0]=dp[i-1][1];
		}
	}
	for(int i=1;i<=n;i++)
	{
		ans=ans+dp[i][1];
	}
	printf("%d",ans);
	return 0;
}
2023/7/17 13:50
加载中...