POJ - 2886 Who Gets the Most Candies?(线段树)

来源：互联网发布：开源网络管理软件编辑：程序博客网时间：2024/06/05 23:01

既然是做线段树专题，这个题思路很容易有，用线段树维护一个区间和，表示这个区间内还有多少个人，更新是单点增减。

自己做的时候不知道反素数的概念所以就打个表先预处理一下。

参考题解

if(p[pos].v > 0)    k = ((k + p[pos].v - 2) % tree[1].cnt + tree[1].cnt) % tree[1].cnt + 1;else    k = ((k + p[pos].v - 1) % tree[1].cnt + tree[1].cnt) % tree[1].cnt + 1;

从这篇题解里，利用取余运算，避免了各种讨论，我应该好好学习。

#include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>using namespace std;#define lson l,mid,rt<<1#define rson mid+1,r,rt<<1|1const int maxn=500000+100;int sum[maxn<<2], cnt[maxn], A[maxn], id[maxn];char name[maxn][12];void init(){    for(int i=1; i<maxn; i++)        for(int j=i; j<maxn; j+=i)            ++cnt[j];    for(int i=1; i<maxn; i++)        id[i] = cnt[id[i-1]] >= cnt[i] ? id[i-1] : i;}void pushup(int rt){    sum[rt]=sum[rt<<1]+sum[rt<<1|1];}void build(int l, int r, int rt){    sum[rt]=1;    if(l==r)return ;    int mid=(l+r)>>1;    build(lson);    build(rson);    pushup(rt);}int update(int p, int l, int r, int rt){    sum[rt]--;    if(l==r)return l;    int mid=(l+r)>>1;    if(sum[rt<<1] >= p)return update(p, lson);    else return update(p-sum[rt<<1], rson);}int main(){    init();    int n, k, op;    while(~scanf("%d%d", &n, &k))    {        for(int i=1; i<=n; i++)            scanf("%s%d", name[i], &A[i]);        build(1, n, 1);        int pos=0;        A[0]=0;        for(int i=0; i<id[n]; i++)        {            if(A[pos]>0)                k=((k+A[pos]-2)%sum[1]+sum[1])%sum[1]+1;            else k=((k+A[pos]-1)%sum[1]+sum[1])%sum[1]+1;            pos=update(k, 1, n, 1);        }        printf("%s %d\n", name[pos], cnt[id[n]]);    }    return 0;}

0 0