CF1213E Two Small Strings

题目链接

问题分析

由于三个字母是等价的，所以大致可以分为如下几种情况：

• aa, ab

• ab, ac

• ab, ba

• ab, bc

不难发现，第\(3\)中情况可能造成无解（\(n>1\)时），而剩下的情况都可以由\(aaabbbccc\)或\(abcabcabc\)这样的串解决。所以直接枚举\(3\)个字母的全排列，然后拓展成上面两种情况分别判断一下即可。

参考程序

#include <bits/stdc++.h>
using namespace std;

int n;
char A[ 10 ], B[ 10 ];
int Map[ 10 ][ 10 ];
int Ans[ 10 ];

bool Check1() {
    return Map[ Ans[ 1 ] ][ Ans[ 2 ] ] || Map[ Ans[ 2 ] ][ Ans[ 3 ] ] || Map[ 1 ][ 1 ] || Map[ 2 ][ 2 ]|| Map[ 3 ][ 3 ];
}

bool Check2() {
    return Map[ Ans[ 1 ] ][ Ans[ 2 ] ] || Map[ Ans[ 2 ] ][ Ans[ 3 ] ] || Map[ Ans[ 3 ] ][ Ans[ 1 ] ];
}

int main() {
    scanf( "%d", &n );
    scanf( "%s%s", A + 1, B + 1 );
    Map[ A[ 1 ] - 'a' + 1 ][ A[ 2 ] - 'a' + 1 ] = 1;
    Map[ B[ 1 ] - 'a' + 1 ][ B[ 2 ] - 'a' + 1 ] = 1;
    Ans[ 1 ] = 1; Ans[ 2 ] = 2; Ans[ 3 ] = 3;
    if( n == 1 ) {
        while( Map[ Ans[ 1 ] ][ Ans[ 2 ] ] || Map[ Ans[ 2 ] ][ Ans[ 3 ] ] )
            if( !next_permutation( Ans + 1, Ans + 4 ) ) {
                printf( "NO\n" );
                return 0;
            }
        printf( "YES\n" );
        printf( "%c%c%c\n", Ans[ 1 ] + 'a' - 1, Ans[ 2 ] + 'a' - 1, Ans[ 3 ] + 'a' - 1 );
        return 0;
    }
    while( true ) {
        if( !Check1() ) {
            printf( "YES\n" );
            for( int i = 1; i <= n; ++i ) printf( "%c", Ans[ 1 ] + 'a' - 1 );
            for( int i = 1; i <= n; ++i ) printf( "%c", Ans[ 2 ] + 'a' - 1 );
            for( int i = 1; i <= n; ++i ) printf( "%c", Ans[ 3 ] + 'a' - 1 );
            printf( "\n" );
            return 0;
        }
        if( !Check2() ) {
            printf( "YES\n" );
            for( int i = 1; i <= n; ++i )
                printf( "%c%c%c", Ans[ 1 ] + 'a' - 1, Ans[ 2 ] + 'a' - 1, Ans[ 3 ] + 'a' - 1 ) ;
            printf( "\n" );
            return 0;
        }
        if( !next_permutation( Ans + 1, Ans + 4 ) ) {
            printf( "NO\n" );
            return 0;
        }
    }
    return 0;
}