首页 > 其他 > 详细

HackerRank - "Identify Smith Numbers"

时间:2015-05-28 01:58:03      阅读:216      评论:0      收藏:0      [点我收藏+]

I guess the punch line of this one is Sieving for primes.

#include <cmath>
#include <cstdio>
#include <climits>
#include <cctype>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

vector<int> primes;

int digiSum(int n)
{
    int ret = 0;
    while (n)
    {
        ret += n % 10;
        n /= 10;
    }
    return ret;
}

int facSum(int n)
{
    int ret = 0;
    
    int pinx = 0;
    while (n > 1 && pinx < primes.size())
    {
        int p = primes[pinx];
        while (n % p == 0)
        {
            n /= p;
            ret += digiSum(p);
        }
        pinx++;
    }
    if (n > 1)
    {
        ret += digiSum(n);
    }
    return ret;
}

void go(int n)
{
    int dSum = digiSum(n);
    int fSum = facSum(n);
    cout << (dSum == fSum ? 1 : 0) << endl;
}

void sieving(int n)
{
    int bound = ceil(sqrt(n));
    vector<bool> mark(bound + 1, true);

    int removed = 1;
    int picked = 2;
    while (removed)
    {
        removed = 0;
        for (int i = 2 * picked; i <= bound; i += picked)
        {
            mark[i] = false;
            removed++;
        }

        picked++;
        while (!mark[picked]) picked++;
    }    

    for (int i = 2; i < mark.size(); i ++)
        if (mark[i])
            primes.push_back(i);
}

int main() 
{
    sieving(INT_MAX);
    int n; cin >> n;
    go(n);
    return 0;
}

HackerRank - "Identify Smith Numbers"

原文:http://www.cnblogs.com/tonix/p/4534844.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!