NOIP 模拟赛 nan （暴力？）

思路

这道题。。。我找不到匹配的算法。。。
刚开始想打表，发现时间空间都要炸。然后准备推个公式，发现不好推。
回归打表大暴力。选择分块打表，按照数字的个数分块，发现剩下的都是等差数列。记住，加一下%一下，开long long不要爆了。
然后查找的时候二分一下，效率efficiency。
打表的话，好像只要60、70W就好了。

代码

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#include<cmath>#include<vector>using namespace std;const int mod=1000000007;const int N=1400000+10;int T,a[N];long long sum[N],pre[N];void init() {    a[1]=1;a[2]=a[3]=2;    long long last=3,cur=3;    while(true)     {        for (int i=last+1;i<=last+a[cur];i++) a[i]=cur;        last+=a[cur];        cur++;        if (last+a[cur]>N) break;    }    for (int i=1;i<=N;i++) sum[i]=sum[i-1]+a[i];    last=0;    for (int i=1;i<=N;i++)     {        long long L=last+1,R=last+a[i];        pre[i]=((long long)a[i]*((long long)L+R)>>1)%mod*i%mod;        (pre[i]+=pre[i-1])%=mod;        last=R;    }}int main() {    freopen("nan.in","r",stdin);    freopen("nan.out","w",stdout);    init();    scanf("%d",&T);    while(T--)     {        int n;        scanf("%d",&n);        int pos=upper_bound(sum+1,sum+N+1,n)-sum-1;        long long ans=pre[pos];        if(sum[pos]<n)         {            long long len,L,R;            len=n-sum[pos];            L=sum[pos]+1;            R=sum[pos]+len;            pos++;            ans+=((long long)len*((long long)L+R)>>1)%mod*pos%mod;            ans%=mod;        }        printf("%I64d\n",ans);    }    return 0;}