码迷,mamicode.com
首页 > 其他好文 > 详细

HNU13303 Counting substhreengs(递推)

时间:2017-05-31 14:06:43      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:字符串   sub   strlen   int   style   include   art   ack   targe   

题目:http://acm.hnu.cn/online/?

action=problem&type=show&id=13303&courseid=0

题意:给你一个字符串,由数字和其它字符组成,问有多少个子串,使得子串里面的数字和为3的整数倍(子串必须连续,并且里面不能有其它字符)。

分析:开3个数组,dp0[i],dp1[i],dp2[i]。dp0[i]表示从字符串的最后一个位置到位置i有多少个字符串的和是0的倍数,dp1[i],dp2[i]类似。

从后往前推,那么假如当前位置为i,(str[i]-‘0‘)%3==1,dp0[i]=dp2[i+1],。。。。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e6+10;
char str[maxn];
long long dp0[maxn],dp1[maxn],dp2[maxn];
int main()
{
	int len,i,j,x;
	long long ans,Sum;
	while(scanf("%s",str)!=EOF)
	{
		len=strlen(str);
		memset(dp0,0,sizeof(dp0[0])*(len+2));
		memset(dp1,0,sizeof(dp0[0])*(len+2));
		memset(dp2,0,sizeof(dp0[0])*(len+2));
		ans=0;
		for(i=len-1;i>=0;i--)
		{
			if(str[i]>=‘0‘ && str[i]<=‘9‘)
			{
				x=(str[i]-‘0‘)%3;
				if(x==1)
				{
					ans+=dp2[i+1];
					dp0[i]=dp2[i+1];
					dp1[i]=dp0[i+1];
					dp2[i]=dp1[i+1];
					dp1[i]++;
				}
				else if(x==2)
				{
					ans+=dp1[i+1];
					dp0[i]=dp1[i+1];
					dp1[i]=dp2[i+1];
					dp2[i]=dp0[i+1];
					dp2[i]++;
				}
				else
				{
					ans+=dp0[i+1]+1;
					dp0[i]=dp0[i+1];
					dp1[i]=dp1[i+1];
					dp2[i]=dp2[i+1];
					dp0[i]++;
				}
			}
			else
			{
				dp0[i]=dp1[i]=dp2[i]=0;
			}
		}
		cout<<ans<<‘\n‘;
	}
	return 0;
}


HNU13303 Counting substhreengs(递推)

标签:字符串   sub   strlen   int   style   include   art   ack   targe   

原文地址:http://www.cnblogs.com/claireyuancy/p/6923459.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!