首页 > 其他 > 详细

HackerRank - "Red John is Back"

时间:2015-07-03 07:56:44      阅读:238      评论:0      收藏:0      [点我收藏+]

1AC. 1D DP + Sieving

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

int ways(int n)
{
    if (n <= 1) return n;

    vector<int> dp(n + 1, 0);
    dp[0] = 1;
    for (int i = 0; i < n; i++)
    {
        dp[i + 1] += dp[i];
        if ((i + 4) <= n) dp[i + 4] += dp[i];
    }
    return dp[n];
}

int cntPrimes(int n) // sieving
{
    vector<int> mark(n, 1);
    mark[0] = 0;

    int inx = 1; // 2
    int scnt = 1;
    while (inx < n && scnt > 0)
    {
        scnt = 0;
        for (int i = 2* inx + 1; i < n; i += (inx + 1))
        {
            mark[i] = 0;
            scnt++;
        }
        if (scnt > 0)
        {
            inx++;
            while (inx < n && mark[inx] == 0) inx++;
        }
    }

    return std::accumulate(mark.begin(), mark.end(), 0);
}

int main()
{
    int t; cin >> t;
    while (t--)
    {
        int n; cin >> n;
        int nCnt = ways(n);
        int ret = cntPrimes(nCnt);
        cout << ret << endl;
    }
    return 0;
}

HackerRank - "Red John is Back"

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

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