JZOJ5481. 【清华冬令营2018模拟】第二题

Description

sl又做白日梦了。他梦到若干年后的一天，他成为了地球最高统治者。地球上共有n个城市，被n条双向公路连成一个环，第i条公路连接i-1号城市和i号城市（第一条连接n号城市和1号城市），第i条公路的长度为d[i]。sl曾受到食堂饭菜的摧残，而且他认为食堂饭菜之所以这么难吃，是因为大家可以在外面吃到美味的饭菜，突出了食堂饭菜的难吃。所以他决定在这n个城市中建一个公共食堂，所有人都只能吃这个食堂里的饭菜。假设食堂建在i号城市，那么j号城市需要将饭菜从i号城市运到j号城市，设从i到j的两条路径长度分别为l1,l2，j号城市对饭菜的需求为w[j]，运费则为l1*l2*w[j]。sl会选一个让总运费最少的城市建食堂，如果有多个，他会等概率选一个。
1～n-1号城市的w已经知道了，但n号城市不太支持sl的统治，所以sl只知道它的w在实数区间[a,b]内等概率分布。所以他想知道他在每个城市建食堂的概率是多少。

Input

第一行三个正整数n、a、b。
接下来n-1行每行一个正实数，为w[1]到w[n-1]。
接下来n行每行一个正实数，为d[1]到d[n]。

Output

n行，每行一个实数，第i行的实数为在i号城市建食堂的概率。
绝对误差在0.001以内为正确。

Sample Input

5 1 100
50
25
25
50
1
2
3
2
1

Sample Output

0.090
0.000
0.000
0.090
0.821

Data Constraint

对于20%的数据，n≤1000。
对于另20%的数据，a=b。
对于另20%的数据，w[1]=w[2]=……=w[n-1]

题解

通过题目可以知道，
每个位置最终的总运费是关于$w_n$的一个一次函数。
斜率是l1*l2，截距就是其它点的运费和。
一次函数的图像直线。
所以在点n建饭堂的总运费就是一条水平的直线，其余都是倾斜的直线。
最优答案就是在这些直线中选取最下面的一条作为答案。
很显然维护一个凸壳，然后就分段求答案。

关键问题是如何求出每一条直线，
如果是$O(n^2)$的复杂度，只有20分。

观察式子可以发现，截距$b_i$与$b_{i-1}$有很多相同的地方，
故此想到作差。
维护几个前缀和，这样值需要求出$b_1$，
然后其它点就是可以O(1)计算了。

维护凸壳也很简单，
就先让其中一个东西有序，比如斜率，
再按照用一个栈来维护。

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#define ll long long
#define N 100003
#define db double
#define P putchar
#define G getchar
#define mo 998244353
using namespace std;
char ch;

db max(db a,db b){return a>b?a:b;}
db min(db a,db b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

struct node
{
    db f,g;
    int z;
}q[N];

int n,z[N],top,t;
db w[N],d[N],s[N*2],sum,f[N],g[N],len,sw,len1;
db ans[N],t1,t2,su[N],a,b;

db get(int x,int y) 
{
    if(x>y)swap(x,y);
    return (s[y]-s[x])*(sum-(s[y]-s[x]));
}

db jd(int x,int y)
{
    return(db)(q[y].f-q[x].f)*1.000/(q[x].g-q[y].g);
} 

bool pd(int x,int y,int z)
{
    return(jd(x,z)<=jd(x,y));
}

bool cmp(node a,node b)
{
    return a.g>b.g;
}

int main()
{
    freopen("dierti.in","r",stdin);
    freopen("dierti.out","w",stdout);
    scanf("%d%lf%lf",&n,&a,&b); 
    if(a==b)b+=0.000000001;
    for(int i=1;i<n;i++)
        scanf("%lf",&w[i]),sw=sw+w[i],g[i]=g[i-1]+w[i];
    for(int i=1;i<=n;i++)
        scanf("%lf",&d[i]),s[i]=s[i-1]+d[i],sum+=d[i],f[i]=f[i-1]+s[i]*w[i];

    q[1].z=1;
    su[1]=1;
    for(int j=1;j<n;j++)
        q[1].f+=get(1,j)*w[j];
    q[1].g=get(1,n);

    for(int i=2;i<n;i++)
    { 
        q[i].z=i;
        su[i]=1;

        q[i].f=q[i-1].f;
        q[i].f+=d[i]*(sum-d[i])*(w[i-1]-w[i]);
        q[i].f+=d[i]*d[i]*(sw-w[i]-w[i-1]);
        q[i].f+=d[i]*sum*(g[i-2]-g[n-1]+g[i]);
        q[i].f+=2*d[i]*(w[i-1]*d[i]-s[i]*g[n-1]+f[n-1]);

        q[i].g=get(i,n);
    } 
    su[n]=1;
    q[n].z=n;
    for(int j=1;j<n;j++)
        q[n].f+=get(n,j)*w[j];


    sort(q+1,q+1+n,cmp);

    z[top=1]=1;
    for(int i=2;i<=n;i++)
    {
        if(q[i-1].f==q[i].f && q[i-1].g==q[i-1].g)
        {
            if(ans[q[i-1].z]<0)
            {
                t=floor(-ans[q[i-1].z]);
                su[t]++;
                ans[q[i].z]=ans[q[i-1].z];
            }
            else
            {
                su[q[i-1].z]++;
                ans[q[i].z]=-q[i-1].z;
            }
            continue;
        }
        while(top>1 && pd(z[top-1],z[top],i))top--;
        z[++top]=i;
    }

    t1=a;
    t2=jd(z[1],z[2]);
    t2=(t2>b?b:t2);
    if(t1<=t2)ans[q[z[1]].z]=(t2-t1)/(db)(b-a);

    t1=jd(z[top],z[top-1]);
    t2=b;
    t1=(t1<a?a:t1);
    if(t1<=t2)ans[q[z[top]].z]=(t2-t1)/(db)(b-a);

    for(int i=2;i<top;i++)
    {
        t1=jd(z[i],z[i-1]);
        t2=jd(z[i],z[i+1]);
        t1=(t1<a?a:t1);
        t2=(t2>b?b:t2);
        if(t1<=t2)ans[q[z[i]].z]=(t2-t1)/(db)(b-a);
    }

    for(int i=1;i<=n;i++)
        if(ans[i]<0)t=floor(-ans[i]),ans[i]=ans[t]/su[t];

    for(int i=1;i<=n;i++)
        ans[i]=ans[i]/su[i];

    for(int i=1;i<=n;i++)
        printf("%.3lf\n",ans[i]);
}