LightOJ 1097 - Lucky Number 线段树

http://www.lightoj.com/volume_showproblem.php?problem=1097

题意：一个自然数序列，先去掉所有偶数项，在此基础上的序列的第二项为3，则删去所有3的倍数的元素，再是7……重复操作，最后问第n项的值

思路：使用线段树构造序列，对一个数进行标记是否已被删去，和为元素个数。由于样例给出了大小，所以很容易控制空间。

/** @Date    : 2016-12-05-19.34

  * @Author  : Lweleth (SoungEarlf@gmail.com)

  * @Link    : https://github.com/

  * @Version :

  */

#include<bits/stdc++.h>

#define LL long long

#define PII pair

#define MP(x, y) make_pair((x),(y))

#define fi first

#define se second

#define PB(x) push_back((x))

#define MMG(x) memset((x), -1,sizeof(x))

#define MMF(x) memset((x),0,sizeof(x))

#define MMI(x) memset((x), INF, sizeof(x))

using namespace std;

const int INF = 0x3f3f3f3f;

const int N = 1429431;

const double eps = 1e-8;

struct yuu

{

    int l, r;

    int sum;

}tt[N * 3];

int ans[100010];

void pushup(int p)

{

    tt[p].sum = tt[p << 1].sum + tt[p << 1 | 1].sum;

}

void build(int l, int r, int p)

{

    tt[p].l = l;

    tt[p].r = r;

    if(l == r)

    {

        tt[p].sum = l & 1;

        return ;

    }

    int mid = (l + r) >> 1;

    build(l , mid, p << 1);

    build(mid + 1, r, p << 1 | 1);

    pushup(p);

}

void updata(int x, int p)

{

    if(tt[p].l == tt[p].r)

    {

        tt[p].sum = 0;

        return ;

    }

    if(x <= tt[p << 1].sum)

        updata(x, p << 1);

    else

        updata(x - tt[p << 1].sum, p << 1 | 1);

    pushup(p);

}

int query(int len, int p)

{

    if(tt[p].l == tt[p].r)

        return tt[p].l;

    if(len <= tt[p << 1].sum)

        query(len, p << 1);

    else query(len - tt[p << 1].sum, p << 1 | 1);

}

int main()

{

    int c = 1;

    ans[1] = 1;

    build(1, 1429431, 1);

    for(int i = 2; i <= 1429431; i++)

    {

        int p = query(i, 1);

        for(int j = 0; p + j <= 1429431; j+=p-1)

            updata(p+j, 1);

        if(c >= 100000)

            break;

        ans[++c] = query(i, 1);

    }

    int T;

    int cnt = 0;

    cin >> T;

    while(T--)

    {

        int n;

        cin >> n;

        printf("Case %d: %d\n", ++cnt, query(n, 1));

    }

    return 0;

}

巴特西

LightOJ 1097 - Lucky Number 线段树

最新文章

热门文章