Codeforces Round #215_div2_C. Sereja and Algorithm

转载注明出处  http://blog.csdn.net/moedane

 

 

传送门 http://codeforces.com/contest/368/problem/C

 

题意

给出一个只包含’x’，’y’，’z’的字符串，给出查询区间l和r，问[l,r]内任选出一个长度为3的子串进行重新排列，问最终能否使得[l,r]内任意长度为3的子串都达成"zyx", "xzy", "yxz"这样的形式。

（虽然原题不是这样描述的……但大意是酱紫问题不大）

 

思路

由于可以任选长度为3的子串进行重排，那么只需要保证[l,r]内x,y,z分别出现的次数等于lenth或lenth-1即可，不必在意其初始位置。这里用三个前缀数组即可统计。

其中x,y,z出现的次数，必须要有lenth%3个是等于lenth/3 + 1的，要有（n - lenth%3）个是等于lenth/3的。只有这种情况才能达到要求。

最后要特判，lenth小于3的直接输出YES。

 

代码

#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <cctype>
#define bug puts("here");

using namespace std;

typedef long long ll;

const int maxn = 100086;
const int mod = 1000000007;
const double PI = atan(1.0)*4.0;

char a[maxn];
int x[maxn],y[maxn],z[maxn];

int main()
{
    scanf("%s",a);
    int len = strlen(a);
    int i;
    memset(x,0,sizeof(x));
    memset(y,0,sizeof(y));
    memset(z,0,sizeof(z));
    //做前缀数组
    for(i=1;i<=len;i++)
        x[i] = x[i-1] + (a[i-1] == 'x');
    for(i=1;i<=len;i++)
        y[i] = y[i-1] + (a[i-1] == 'y');
    for(i=1;i<=len;i++)
        z[i] = z[i-1] + (a[i-1] == 'z');
    int m;
    scanf("%d",&m);
    for(i=0;i<m;i++)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        int xx,yy,zz;
        xx = x[r] - x[l-1];//统计当前区间内x，y，z出现的次数
        yy = y[r] - y[l-1];
        zz = z[r] - z[l-1];
        int lenth = r - l + 1;
        if(lenth < 3)
        {
            puts("YES");
            continue;
        }
        int t1,t2;
        t1 = t2 = 0;
        if(xx == lenth/3) t1 ++;
        else if(xx == lenth/3 + 1) t2++;
        if(yy == lenth/3) t1 ++;
        else if(yy == lenth/3 + 1) t2++;
        if(zz == lenth/3) t1 ++;
        else if(zz == lenth/3 + 1) t2++;
        if(t2 == lenth%3 && t1 == 3 - t2) puts("YES");
        else puts("NO");
   }
   return 0;
}