HDU 6098 Inversion（水~）

Description

给出一长度为$n$的序列$A_i$，定义$B_i=\sum\limits_{i\nmid j}A_j,i\ge 2$，求$B_2,...,B_n$

Input

第一行一整数$T$表示用例组数，每组用例输入一整数$n$表示序列长度，之后输入$n$个整数$A_1,...,A_n$

$(1\le T\le 20,2\le n\le 10^5,1\le A_i\le 10^9)$

Output

输出$B_2,...,B_n$

Sample Input

2
4
1 2 3 4
4
1 4 2 3

Sample Output

3 4 3
2 4 4

Solution

对于一个$i$，是其倍数的$j$最多$\lfloor\frac{n}{i}\rfloor$个，把$A$值从大到小排序，对每个$i$，从大到小考虑每个$A$值的编号，如果一个编号不能被$i$整除那么这个值就是$B_i$了，这样最多$\lfloor\frac{n}{i}\rfloor$次就可以求出$B_i$，总时间复杂度$O(nlog_2n)$

Code

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=100001;
int T,n;
struct node
{
    int v,id;
    bool operator<(const node&b)const
    {
        return v<b.v;
    }
}a[maxn];
int b[maxn];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i].v),a[i].id=i;
        sort(a+1,a+n+1);
        for(int i=n;i>=2;i--)
        {
            for(int j=n;j>=1;j--)
                if(a[j].id%i!=0)
                {
                    b[i]=a[j].v;
                    break;
                }
        }
        for(int i=2;i<=n;i++)printf("%d%c",b[i],i==n?'\n':' ');
    }
    return 0;
}