Uva 10042 - Smith Numbers


 // Accepted
// Time: 0.013

// Header file begin
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<string>
#include<vector>
#include<cmath>
#include<cctype>
#include<sstream>
#include<set>
#include<list>
#include<stack>
#include<utility>
#include<queue>
#include<algorithm>
#include<cstdlib>
#include<bitset>
// End
//..........
// Macro
#define sf scanf
#define pf printf
#define lp1(i,n) for(i=0;i<n;i++)
#define lp2(i,n) for(i=1;i<=n;i++)
#define mset(c,v) memset(c,v,sizeof(c))
#define cp(a) cout<<" "<<a<<" "
#define nl puts("")
#define sq(x) ((x)*(x))
#define all(x) x.begin(),x.end()
#define reall(x) x.rbegin(),x.rend()
#define s_wap(x,y) x^=y;y^=x;x^=y;
#define sz size()
#define gc getchar()
#define pb push_back
#define freader freopen("input.txt","r",stdin)
// End.........

// Size
#define mx7 20000100
#define mx6 1500000
#define mx5 100005
#define mx4 1000100
#define inf 1<<30                                           //infinity value
#define eps 1e-9
#define mx (65540)
#define mod 1000000007
#define pi acos(-1.0)

// Macros for Graph

#define white 1
#define gray 2
#define black 3
#define nil 0

using namespace std;
/***************/

// typedef

typedef long long LL;
typedef long L;
typedef unsigned long long ull;
typedef unsigned long ul;
typedef unsigned int ui;
typedef pair<int, int> pii;
typedef vector<int>vi;
typedef vector<long long> vll;
typedef vector<long>vl;
typedef vector<char>vch;
typedef vector<string>vs;
typedef map<int,int>mpii;
typedef map<int,bool>mpbi;
typedef map<long,bool>mpbl;
typedef map<long long,bool>mpbll;
typedef map<char,int>mpci;
typedef map<char,bool>mpbc;
typedef map<string,int>mpsi;
typedef map<long long,long long>mpll;

// template

template<class T> T gcd(T a, T b ) {return b!=0?gcd<T>(b,a%b):a;}
template<class T> T large(T a, T b ) {return a>b?a:b;}
template<class T> T small(T a, T b ) {return a<b?a:b;}
template<class T> T diffrnce(T a, T b) {return a-b<0?b-a:a-b;}

bitset<mx4+9>bs;
int prm[(mx4>>1)+9],plen=0;
LL psz;

void sieve()
{
    LL i,j;
    psz=mx4+1;

    bs.set();
    bs[0]=bs[1]=0;

    for(i=2;i<=psz;i++)
    {
        if(bs[i])
        {
            prm[plen++]=i;

            for(j=i*i;j<=psz;j+=i)
            {
                bs[j]=0;
            }
        }
    }

    //lp1(i,100)cp(prm[i]);
}

bool isPrime(LL n)
{
    if(n <= psz)return bs[n];

    for(int i=0;i<plen and sq(prm[i])<=n; i++)
    {
        if(!(n%prm[i]))return false;
    }

    return true;
}

LL digit_sum(LL n)
{
    LL sum=0;
    while(n)
    {
        sum+=(n%10);
        n/=10;
    }

    return sum;
}

LL spf(LL n)
{
    LL s=0;
    for(int i=0;i<plen and sq(prm[i])<=n; i++)
    {
        if(!(n%prm[i]))
        {
            while(!(n%prm[i]))
            {
                s+=digit_sum(prm[i]);
                n/=prm[i];
                if(n==0 or n==1)break;
            }
        }
    }

    if(n>1)
    {
        s+=digit_sum(n);
    }

    return s;
}

int main()
{
    sieve();
    int t;
    sf("%d",&t);

    while(t--)
    {
        LL num,k,tmp;
        sf("%lld",&num);
        tmp=num;

        for(k=tmp+1;;k++)
        {
            //cout << "k = " << k << ": ";
            LL lft,rgt;

            if(isPrime(k))continue;

            lft = digit_sum(k); //cp(lft);
            rgt = spf(k); //cp(rgt);
            //nl;

            if(lft == rgt)
            {
                pf("%lld\n",k);
                break;
            }
        }
    }

    return 0;
}

Comments

Popular posts from this blog

SPOJ-CMG - Collecting Mango

LightOJ 1009 - Back to Underworld

LeetCode Palindrome Number