HDU 6387 AraBellaC(二分)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/V5ZSQ/article/details/82562840

Description

已知一个由 A , B , C 组成的字符串是由一个较短串为循环串生成,且该较短串为 x A y B z C 依次组成,其中 x , y , z 均为正,现在给出该字符串中 n 个位置的字符,要求找到字典序最小的三元组 ( x , y , z ) 满足所给限制

Input

第一行一整数 T 表示用例组数,每组用例输入一整数 k 和一个字符 ( A / B / C ) 表示 k 位置字符

( 1 T 40 , 1 n 5000 , 1 k 10000 )

Output

如果存在解则输出字典序最小的 ( x , y , z ) ,否则输出 N O

Sample Input

2
3
1 A
2 B
3 C
4
1 A
2 C
3 B
4 C

Sample Output

1 1 1
NO

Solution

枚举循环节长度 l e n ,对于字母 i ,假设其出现的位置为 a i 1 , . . . , a i k ,那么可以二分求出该字母在一个循环中出现的最小位置 m i n i = m i n ( ( a i j 1 ) % l e n ) 和最大位置 m a x i = m a x ( ( a i j 1 ) % l e n ) ,这两个位置构成该字母在一个循环内的最小存在区间,如果这三个区间有交或者某个区间为空则该循环节无解,否则对于该循环, A 出现的最小次数即为 m a x 1 B 出现的最小次数即为 m a x 2 m a x 1 ,而 C 出现的最小次数即为 l e n m a x 2 ,以此更新最优解即可,时间复杂度 O ( T m l o g m ) ,其中 m 为循环节的最大可能长度

Code

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int INF=0x3f3f3f3f;
vector<int>p[3]; 
vector<int>::iterator it;
int T,m,Min[3],Max[3];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&m);
        for(int i=0;i<3;i++)p[i].clear();
        int mx=0;
        while(m--)
        {
            int c;
            char s[3];
            scanf("%d%s",&c,s);
            p[s[0]-'A'].push_back(c);
            mx=max(mx,c);
        }
        for(int i=0;i<3;i++)sort(p[i].begin(),p[i].end());
        int a=INF,b=INF,c=INF;
        for(int len=3;len<=mx+2;len++)
        {
            for(int i=0;i<3;i++)
            {
                Max[i]=0,Min[i]=INF;
                int L=1,R=len+1;
                while(L<=mx)
                {
                    it=lower_bound(p[i].begin(),p[i].end(),L);
                    if(it!=p[i].end()&&(*it)<R)
                    {
                        Min[i]=min(Min[i],(*it)-L+1);
                        it=lower_bound(p[i].begin(),p[i].end(),R);
                        it--;
                        Max[i]=max(Max[i],(*it)-L+1);
                    }
                    L+=len,R+=len;
                }
            }
            if(Max[0]==0)Min[0]=Max[0]=1;
            if(Max[0]==len)continue;
            if(Max[1]==0)Min[1]=Max[1]=Max[0]+1;
            if(Max[1]==len)continue;
            if(Max[2]==0)Min[2]=Max[1]+1,Max[2]=len;
            if(Max[0]>=Min[1]||Max[1]>=Min[2])continue;
            int ta=Max[0],tb=Max[1]-Max[0],tc=len-Max[1];
            if(a>ta||a==ta&&b>tb||a==ta&&b==tb&&c>tc)a=ta,b=tb,c=tc;
        }
        if(a==INF)printf("NO\n");
        else printf("%d %d %d\n",a,b,c);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/V5ZSQ/article/details/82562840