hdu 5439 Aggregated Counting 2015长春网络赛 分块 二分 预处理

题目

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5439

题目来源：2015长春网络赛第八题。

简要题意：给定一个数列，构建方式是开始时$1,2,2$第$i$个为$s_i$就在数列后面加$s_i$个$i(i>2)$，设$n$最后一次出现的位置是$p$，求$p$最后一次出现的位置模个大数。

数据范围：$T\leqslant 2000;\quad n\leqslant 10^9;\quad MOD=10^9+7$

题解

可以说这是一道非常抽象而且巧妙的题目，非常可惜的是赛场上脑子一团乱，没能A出来。

先来假设我们需要的数列吧$s_i$就是该数列，$f_i$为$i$出现的最后一个位置。

也就是说我们现在要求的是$f_{f_n}$了。

考虑$f_i$与$s_i$之间的关系，得到$f_{i}=f_{i-1}+s_i$，后边加上$s_i$个$i$之后就不会再有了。

同时根据$f_i$的定义我们可以知道$s_{f_{i-1}+1}\cdots s_{f_i}$中的数都是$i$，则：

$$\begin{equation} \begin{aligned} f_{f_i}&=f_{f_{i-1}}+\sum^{f_i}_{j=f_{i-1}+1}j\times s_{j}\\ &=f_{f_{i-1}}+i\times\sum^{f_i}_{j=f_{i-1}+1}j\\ &=f_{f_{i-1}}+\frac{i(f_i+f_{i-1}+1)(f_i-f_{i-1})}{2} \end{aligned} \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \qquad \end{equation}$$

两者其实是可以互推的，我们可以先用$s_i$处理出$f_i$然后再用$f_i$去反推$s_i$，然后用$s_i$推出最终的结果。

整个过程非常抽象，需要好好理解，非常巧妙需要强大分析能力的一题。

实现

构建$f_i$的过程实际上就是模拟搞出$s_i$。

然后再用$f_{i}=f_{i-1}+s_i$预处理需要的$f_i$，大概要处理到$440000$左右。

由于$T$比较大，我们可以先预处理出来$f_{f_i}$然后二分$f_i$。

最后剩下一段的一段也用等差公式来求就行了。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 5e5+5;
const LL MOD = 1e9+7;
int s[N];
int f[N];
LL su[N];

LL sum(LL l, LL r, LL x) {
    return (l+1+r)*(r-l)/2*x;
}

LL get(LL x) {
    int pos = upper_bound(f, f+N, x)-f;
    return su[pos-1]+sum(f[pos-1], x, pos);
}

void init() {
    s[1] = 1, s[2] = s[3] = 2;
  f[1] = 1; f[2] = 3;
    for (int i = 3; i < N; i++) {
        f[i] = f[i-1]+s[i];
        for (int j = f[i-1]+1; j <= f[i] && j < N; j++) {
            s[j] = i;
        }
    }
    for (int i = 1; i < N; i++) {
        su[i] = (su[i-1]+sum(f[i-1], f[i], i))%MOD;
    }
}

int main()
{
    init();
    int t, n;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        printf("%I64d\n", get(n)%MOD);
    }
    return 0;
}