HDU 5439 Aggregated Counting

题目大意：

由1开始不断往数组中添加数

就是按照当前所在位置所在的数表示的个数添加这个数目的数

1 2 2 3 3 后面因为要填4，而4号位置为3，说明之后要填3个4

问题就是给定一个n，找到n出现的最后位置p，再找p出现的最后位置即可

这里可以考虑先找到g[i]表示 i 连续出现了多少次

这里想一下的话，因为g[i] 相当于 i 位置出现的数

所以g[i]也满足这个序列

令f[i] 表示 i 出现的最后位置，也就是1~i的总个数

后面去计算g[i]的时候就可以考虑的是找到第 i 个位置在那个f[]的区间内，如果f[k-1]< i <= f[k]

那么说明此时 g[i] = k

那么就可以logn的复杂度计算g[n]了

要计算最后的答案，要考虑的是，给定的n，找到最后出现的p，中间长度 p = g[1]+g[2]....+g[p]

然后再找对应的ans ，那么每次增加的g[i]，就会让整个序列的长度增加 i*g[i]

i*g[i] 可以理解为的是，长度为i的数量有g[i]个，所以总长度是i*g[i]

所以ans = sigma(i*g[i]) i<=n

那么对于n <= 1e9

那么大致计算一下会发现f[500000]>1e9

所以g[n]<500000只要暴力求出前500000的g[] , f[]

那么答案计算前，先找到g[n]是多少

g[n]= lower_bound(f+1 , f+N+1 , n)-f

然后说明[1 , g[n]-1]这一段区间内的所有长度都被用到了

所以之前预处理这个长度的前缀和 sum[]

对于每一个长度 i ，他出现的次数都是 f[i]-f[i-1]

sigma(n*g[n]) f[i-1]<n<=f[i] -> g[n] = i

那么答案就是 i*等差数列了，记得取模(⊙o⊙)哦

然后（g[n]-1 , g[n]]这一段只要枚举（g[n]-1 , n] 就可以了

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <vector>
 5 #include <queue>
 6 using namespace std;
 7 #define M 26
 8 #define N 500000
 9 #define ull unsigned long long
10 #define ll long long
11 const int MOD = 1000000007;
12 int f[N+2] , g[N+2] , cnt[N+2];
13 ll sum[N+2];
14 
15 int Hash(int v)
16 {
17     return lower_bound(f+1 , f+N+1 , v)-f;
18 }
19 
20 void init()
21 {
22     g[1] = 1 , f[1] = 1;
23     g[2] = 2 , f[2] = 3;
24     for(int i=3 ; i<=N ; i++){
25         g[i] = lower_bound(f+1 , f+i , i)-f;
26         f[i] = f[i-1]+g[i];
27     }
28 //    for(int i=1 ;i<=100 ; i++)
29 //        cout<<i<<" "<<g[i]<<" "<<f[i]<<endl;
30 //    cout<<f[N]<<endl;
31 
32     sum[1] = 1;
33     for(int i=2 ; i<=N ; i++){
34         sum[i] = sum[i-1]+(ll)(f[i-1]+1+f[i])*(f[i]-f[i-1])/2 % MOD * (ll)i % MOD;
35       //  if(i<=10) cout<<"sum: "<<i<<" "<<sum[i]<<endl;
36     }
37 }
38 int main() {
39 //    freopen("a.in" , "r" , stdin);
40 //    freopen("out.txt" , "w" , stdout);
41 
42     init();
43     int T , n;
44     scanf("%d" , &T);
45     while(T--){
46         scanf("%d" , &n);
47         int pos = Hash(n);
48         ll ret = sum[pos-1];
49         for(int i=f[pos-1]+1 ; i<=n ; i++) //这个区间每个长度都为pos
50         {
51             ret = (ret+(ll)i*pos)%MOD;
52         }
53         printf("%I64d\n" , ret);
54     }
55     return 0;
56 }