1. 程式人生 > >51NOD 1449——砝碼稱重(貪心演算法)

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<string.h>
#include<stdio.h>
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;
}