Codeforces Round #308 (Div. 2) C. Vanya and Scales dfs

C. Vanya and Scales
Time Limit: 20 Sec
Memory Limit: 256 MB
제목 연결
http://codeforces.com/contest/552/problem/C
Description
Vanya has a scales for weighing loads and weights of masses w0, w1, w2, ..., w100 grams where w is some integer not less than 2 (exactly one weight of each nominal value). Vanya wonders whether he can weight an item with mass m using the given weights, if the weights can be put on both pans of the scales. Formally speaking, your task is to determine whether it is possible to place an item of mass m and some weights on the left pan of the scales, and some weights on the right pan of the scales so that the pans of the scales were in balance.
Input
The first line contains two integers w, m (2 ≤ w ≤ 109, 1 ≤ m ≤ 109) — the number defining the masses of the weights and the mass of the item.
Output
Print word 'YES' if the item can be weighted and 'NO' if it cannot.
Sample Input
3 7
Sample Output
YES
HINT
 
제목
100 개의 분동 을 드 리 겠 습 니 다. i 번 분동 의 질 은 w ^ i 입 니 다. 그리고 m 가 있 는 상황 에서 왼쪽 과 오른쪽 모두 분동 을 놓 아 이 천평 을 균형 있 게 할 수 있 느 냐 고 물 었 습 니 다.
문제 풀이:
dfs 는 그냥 폭력 적 이 었 으 면 좋 겠 는데...
이 분동 에 있어 서 는 세 가지 선택 만 있 을 뿐, 왼쪽 에 놓 고, 놓 지 않 고, 오른쪽 에 놓 아야 한다.
2 와 3 이 yes 를 직접 출력 하기 때문에 답 도 얼마 남지 않 았 다.
코드:
 

#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <typeinfo>
#include <fstream>
#include <map>
#include <stack>
typedef long long ll;
using namespace std;
//freopen("D.in","r",stdin);
//freopen("D.out","w",stdout);
#define sspeed ios_base::sync_with_stdio(0);cin.tie(0)
#define test freopen("test.txt","r",stdin)  
#define maxn 2000001
#define mod 10007
#define eps 1e-5
const int inf=0x3f3f3f3f;
const ll infll = 0x3f3f3f3f3f3f3f3fLL;
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//**************************************************************************************
ll w,m;
int flag;
void dfs(ll a,ll b,ll c)
{

    if(c==m)
    {
        flag=1;
        return;
    }
    if(flag||a==-1)
        return;
    if(abs(b)<=m*w)
    {
        dfs(1,b*w,c+b);
        dfs(1,b*w,c-b);
        dfs(1,b*w,c);
    }

    return;
}
int main()
{
    flag=0;
    w=read(),m=read();
    if(w<=3)
    {
        cout<<"YES"<<endl;
        return 0;
    }
    dfs(1,1,0);
    if(flag)
        cout<<"YES"<<endl;
    else
        cout<<"NO"<<endl;
}