题目大意:规定a1=1,ai表示数字i在i-1出现后出现了几次,例如前几项是1,2,2,3,3,4,4,4,5,5,5,6,6,6,6,etc,求an,n<=1e18。
题解:
直接求是O(n)的,然后每次把i写ai遍可以做到1e10~1e11左右,继续迭代,每次将一段连续的数字写成一段类似等差数列即可。实际算一下大概就是1e7的复杂度。
(代码好短的说)
#include<bits/stdc++.h>
#define lint long long
struct P{lint r,b;}p[11000000];
int main()
{
lint n;scanf("%lld",&n);
if(n==1) return !printf("1\n");
if(n==2) return !printf("2\n");
if(n==3) return !printf("2\n");
p[0].r=3,p[0].b=2,p[1].r=5,p[1].b=3;int i,c=1;
for(i=1;p[c].r<n;i++)
{
lint t=(p[i].r-p[i-1].r)/(p[i].b-p[i-1].b),j;
for(j=1;j<=p[i].b-p[i-1].b&&p[c].r<n;j++)
c++,p[c].r=p[c-1].r+t*(j+p[i-1].b),p[c].b=p[i-1].r+j*t;
}
lint lr=p[c].r-p[c-1].r,ab=p[c].b-p[c-1].b;
return !printf("%lld\n",(n-p[c-1].r+1)/(lr/ab)+p[c-1].b+1);
}