51NOD 1449——砝码称重(贪心算法)

题目如下:

现在有好多种砝码,他们的重量是 w0,w1,w2,...  每种各一个。问用这些砝码能不能表示一个重量为m的东西。

样例解释:可以将重物和3放到一个托盘中,9和1放到另外一个托盘中。


Input
单组测试数据。
第一行有两个整数w,m (2 ≤ w ≤ 10^9, 1 ≤ m ≤ 10^9)。
Output
如果能,输出YES,否则输出NO。
Input示例
3 7
Output示例
YES

分析:题目的意思也就是一个数m,能不能经由多个形如000010000的w进制数加减得到。

从最末位开始往前递推,最后一位只有两种情况,0+1=1,或者0-1=w-1,如果不是这两种情况就是错的。+1或者-1再一步步往前递推便可。



代码如下:

#include
#include
int main()
{
    long long w,m;
    scanf("%lld%lld",&w,&m);
    if(w<=3)
    {
        printf("YES\n");
        return 0;
    }
    while(m)
    {
        if(!((m-1)%w))
            m--;
        else if(!((m+1)%w))
            m++;
        else if(m%w)
        {
            printf("NO\n");
            return 0;
        }
        m=m/w;
    }
    printf("YES\n");
    return 0;
}


你可能感兴趣的:(算法,贪心算法,算法,贪心算法,51NOD)