BZOJ 1011: [HNOI2008]遥远的行星

BZOJ 1011: [HNOI2008]遥远的行星

---

题目

Time Limit: 10 Sec Memory Limit: 162 MB Special Judge

Description

　　直线上N颗行星，X=i处有行星i,行星J受到行星I的作用力，当且仅当i<=AJ.此时J受到作用力的大小为 Fi->j=
Mi*Mj/(j-i) 其中A为很小的常量，故直观上说每颗行星都只受到距离遥远的行星的作用。请计算每颗行星的受力，只要结果的相对误差不超过5%即可.

Input

　　第一行两个整数N和A. 1<=N<=10^5.0.01< a < =0.35,接下来N行输入N个行星的质量Mi,保证0<=Mi<=10^7

Output

　　N行，依次输出各行星的受力情况

Sample Input

5 0.3
3
5
6
2
4

Sample Output

0.000000
0.000000
0.000000
1.968750
2.976000

HINT

　　精确结果应该为0 0 0 2 3,但样例输出的结果误差不超过5%,也算对
　　

---

题解

这道题目如果就单纯模拟时间复杂度是O（an²），但是只要答案的相对误差不超过5%就算正确，所以就不需要很精确

星球j的所受引力为∑(m[k]∗m[j])/(j−k) (k∈[1,i]，i为离j最近且对星球j有作用力的星球)，当j很大的时候，星球所受的引力越接近∑(m[k]∗m[j])/(j−i/2) (k∈[1,i]，i为离j最近且对星球j有作用力的星球)，再用前缀和将引力化为（sum[i]*m[j])/（j-i/2）

---

代码

#include <cstdio>
using namespace std;
double m[100005], sum[100005];
int main()
{
    int n;
    double a,ans;
    scanf("%d%lf",&n,&a);
    for(int j=1;j<=n;j++)
    {
        int i=(int)(a*j+1e-8);
        scanf("%lf",m+j);
        ans=0;
        if(j<=1000)
            for(int k =1;k<=i;k++)
                ans+=m[k]*m[j]/(j-k);
        else
            ans=sum[i]*m[j]/(j-i/2);
        printf("%f\n",ans);
        sum[j]=sum[j-1]+m[j];
    }
    return 0;
}