51nod 1449 砝码称重

贪心算法 CodeForces 修改 隐藏话题
1449 砝码称重
题目来源： CodeForces
基准时间限制：1 秒 空间限制：131072 KB 分值: 40 难度：4级算法题 收藏 关注
现在有好多种砝码，他们的重量是 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

https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1449

这是一个新的姿势。
进制的用法
因为每个砝码只有一个。
所以 可以认为它的砝码加起来是w进制全部为1

我们寻找能不能找到称出m的方法
取决于m是否能和一个只有01的w进制数组成一个新的只有01的w进制数

这样就简单了
我们只需要试着和m的w进制进行匹配。
如果位置为1或者为0 不需要匹配
因为这个地方代表是否要放一个w^n的砝码
之后 如果碰到位置为w-1 的话 证明这里需要放一个w^n的砝码去凑下一位的0或者1
如果这个位置不符合上面两个条件
就证明这个m称不住来
因为，如果不是w-1 证明需要方的砝码数量不止一个
这样就不符合每个砝码只有一个的前提了

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        int z=1;
        while(m)
        {
            if(m%n==1||m%n==0) m/=n;
            else if(m%n==n-1) m=m/n+1;
            else
            {
                z=0;
                break;
            }
        }
        if(z) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}

以上是ac代码 。
当然 较真的我们也可以去找到这个方案 。就是砝码应该怎么放

#include <bits/stdc++.h>
using namespace std;
int tag[10000];
int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        memset(tag,0,sizeof(tag));
        int z=1;
        int t=0;
        while(m)
        {
            if(m%n==1||m%n==0) m/=n;
            else if(m%n==n-1)
            {
                m=m/n+1;
                tag[t]=1;
            }
            else
            {
                z=0;
                break;
            }
            t++;
        }
        if(z) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
        if(z)
        {
            for(int i=0;i<1000;i++)
            {
                if(tag[i]) cout<<i<<endl;
            }
        }
    }
    return 0;
}

记录放了哪些砝码；