很高兴在完全没有参考任何代码的前提下通过此题,呵呵,就是代码比较猥琐,跑得也比较慢,2700MS,超过时限的一半了。
此题应该还有继续提升的空间,我想了想,insert函数和query其实是可以放在一起的。另外网上的方法用了反素数 这样可以减少插入的次数,应该也能剪去一下时间。下次试试。^_^
顺便用此题做为POJ 500题纪念
#include<iostream>
using namespace std;
int n,k;
struct person
{
char s[10];
int p;
}a[500010];
int dp[500010];
void init()
{
int i,j;
for(i=1;i<=500000;i++)
for(j=1;i*j<=500000;j++)
dp[i*j]++;
}
const int maxn=500010;
struct node
{
int l,r;
int cnt;
}tree[maxn*3];
void build(int k,int l,int r)
{
tree[k].l=l;tree[k].r=r;
tree[k].cnt=0;
if(l==r) return;
int mid=(l+r)>>1;
build(k*2,l,mid);
build(k*2+1,mid+1,r);
}
void insert(int i,int k)
{
tree[i].cnt++;
if(tree[i].l==tree[i].r) return;
else
{
int mid=(tree[i].l+tree[i].r)>>1;
if(k<=mid)
insert(i*2,k);
else
insert(i*2+1,k);
}
}
int query(int i,int k)
{
if(tree[i].cnt==0&&tree[i].r-tree[i].l+1==k)
return tree[i].r;
int l=tree[i*2].r-tree[i*2].l+1-tree[i*2].cnt;
int r=tree[i*2+1].r-tree[i*2+1].l+1-tree[i*2+1].cnt;
if(k<=l)
return query(i*2,k);
else
return query(i*2+1,k-l);
}
int query2(int i,int l,int r)
{
if(tree[i].l==l&&tree[i].r==r)
return tree[i].r-tree[i].l+1-tree[i].cnt;
int mid=(tree[i].l+tree[i].r)>>1;
int res=0;
if(r<=mid)
res+=query2(i*2,l,r);
else if(l>mid)
res+=query2(i*2+1,l,r);
else
{
res+=query2(i*2,l,mid);
res+=query2(i*2+1,mid+1,r);
}
return res;
}
int main()
{
int i,j;
init();
int res=dp[1];
int mark=k;
while(scanf("%d%d",&n,&k)!=EOF)
{
int pos;
int res=dp[1];
int mark=k;
build(1,1,n);
for(i=1;i<=n;i++)
scanf("%s%d",a[i].s,&a[i].p);
int l,r;
int t=k;
pos=k;
insert(1,k);
for(i=2;i<=n;i++)
{
t=pos;
if(t>1)l=query2(1,1,t-1);
else l=0;
if(t<n) r=query2(1,t+1,n);
else r=0;
if(a[t].p%(l+r)!=0)
a[t].p%=(l+r);
else if(a[t].p>0)
a[t].p=l+r;
else
a[t].p=-(l+r);
if(a[t].p>0&&a[t].p<=r)
pos=query(1,l+a[t].p);
else if(a[t].p<0&&l+a[t].p>=0)
pos=query(1,l+a[t].p+1);
else if(a[t].p>0&&a[t].p>r)
pos=query(1,a[t].p-r);
else
pos=query(1,l+r-abs(l+a[t].p)+1);
if(res<dp[i])
{
mark=pos;
res=dp[i];
}
insert(1,pos);
}
printf("%s %d\n",a[mark].s,res);
}
return 0;
}