codeforces 451D Count Good Substrings 枚举

   一个只包含a,b字符串,可以将连续若干个相同的字符压缩成一位,比如abaaabb -> abab,如果一个串压缩之后是一个回文串,那么这个串就是一个good string,现在求给出的串的所有子串里,长度为偶数和奇数的good string各有多少个...注意到如果一个串的首尾是相同的,那么这个串一定是一个good string,首先把头尾重复的串压缩成一位(假设是a),那么所夹的一个是一个两端为b的子串,进一步压缩,直到为空,a,b是交替并且成对出现的,所以一定是回文串,那么只要枚举原串的每一位,同时记录一下它之前与他相同的字符在奇数位置,和偶数位置出现的次数,在根据当前位置的奇偶性,就可以累加出最后的答案了。

   

import java.math.BigInteger;  
import java.util.*;
import java.math.*;
public class aaa {

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub		
		Scanner cin=new Scanner(System.in);
		String s;
		int a1,a2,b1,b2;
		long ans1,ans2;
		while(cin.hasNext())
		{
			s=cin.next();
			//System.out.println(s);
			ans1=ans2=0;
			a1=a2=b1=b2=0;
			ans1=s.length();
			for (int i=0; i<s.length(); i++)
			{
				if (s.charAt(i)=='a')
				{
					if (i%2==0)
					{
						ans1+=a2;
						ans2+=a1;
					}
					else 
					{
						ans1+=a1;
						ans2+=a2;
					}
					if (i%2==0) a2++;
					else a1++;
				}
				else 
				{
					if (i%2==0)
					{
						ans1+=b2;
						ans2+=b1;
					}
					else 
					{
						ans1+=b1;
						ans2+=b2;
					}
					if (i%2==0) b2++;
					else b1++;
				}
			}
			System.out.printf("%d %d\n",ans2,ans1);
			//System.out.printf("%I64d %I64d\n",ans2,ans1);
			
		}
        

	}

}

你可能感兴趣的:(codeforces 451D Count Good Substrings 枚举)