51nod1449 砝码称重

Description

现在有好多种砝码，他们的重量是 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

Solution

发现网上的很多题解都写的不够好，也可能是我智商太低看不懂，总之想了很久才想明白。
考虑把m转换成w进制，假设m放在左盘，那么m的质量就等于右盘砝码质量减去左盘砝码，而砝码质量很显然只由0,1组成（w进制下），那么对应地，相减所得的m只会由0和1和w-1组成。
考虑w-1的情况，这种情况在减法时是需要借位的，那么后一位只能由w-1，或者0组成，因此这种情况需要特殊处理一下。我写的时候把m加了一，这样子就可以纳入到前面的情况，比较好做。

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
int w,m;
using namespace std;
int main(){
    scanf("%d%d",&w,&m);
    while (m) {
        if (m%w==1||m%w==0) m=m/w;
          else if (m%w==w-1) m=m/w+1; else {

            printf("NO\n");
            return 0;
          }
    }
    printf("YES\n");
}