数论 - Funny scales(SPOJ - SCALE)

Funny scales 

Problem's Link

 ----------------------------------------------------------------------------

Mean: 

给定两个数n和x，有一个天平，初始时左盘为x，你需要从以下集合中选一些数字来放到两个盘中，使得两个盘相等（note:每个数字只能取一次）。

analyse:

即:

 将X化为3进制：

但是题目说每个3^i必须为1，所以我们需要将X表示成的等式的每一项的系数变为1，这就是本题的关键。

方法：

• 将X化为3进制的过程中，每一项的集合为{0,1,2} ，当遇到2时，将2表示成3-1的形式，即：向前进移位，本位置位-1。

如此一来便很好的处理了系数重复的问题，而且最后统计答案时，根据系数的符号来划分集合即可。

Time complexity: O(N)

 

view code

#include <cstdio>
const int MAXL = 100;

int N , X;
int A [ MAXL ], lenA;
int B [ MAXL ], lenB;
int dig [ MAXL ];

int main()
{

    scanf( "%d %d" , &N , & X );

    for ( int i = 0; X != 0; i ++ )
    {
        dig [ i ] += X % 3;
        if ( dig [ i ] > 1 )
        {
            dig [ i + 1 ] ++;
            dig [ i ] = dig [ i ] - 3;
        }
        X /= 3;
    }

    for ( int i = MAXL - 1; i >= 0; i -- )
    if ( dig [ i ] != 0 )
        if ( dig [ i ] == - 1 )
            A [ lenA ++ ] = i + 1;
        else if ( dig [ i ] == + 1 )
           B [ lenB ++ ] = i + 1;

    if ( A [ 0 ] > N || B [ 0 ] > N )
        printf( "-1 \n " );
    else
    {
        for ( int i = lenA - 1; i >= 0; i -- )
        printf( i ? "%d " : "%d" , A [ i ] );
        printf( " \n " );
        for ( int i = lenB - 1; i >= 0; i -- )
        printf( i ? "%d " : "%d" , B [ i ] );
        printf( " \n " );
    }
    return 0;
}