bzoj4306: 玩具厂

Description

在JIH考察的地图中有N个城市，被公路依次连成了一个环，JIH想在这些城市中建一个玩具厂。城市和公路都被编号为1..N，i号公路连接i-1号城市与i号城市（1号公路连接N号城市与1号城市），每个城市对玩具的需求为wi，每条公路的长度为di。当JIH在第i号城市建玩具厂时，JIH需要将玩具运输到其他城市（当然i城市除外）。设第i号城市到第j号城市的两条路径长度分别为l1、l2，则将玩具运输到第j号城市的费用为l1*l2*wj。总的运输费用为将玩具运到所有城市的运输费用的总和。

JIH当然想要总的运输费用最少，所以他会选最优的城市建玩具厂，如果有多个最优的城市，小月会等概率的选取其中一个建玩具厂。

由于JIH的调查工作没做好，只知道1..N-1号城市的wi，而N号城市的wi只知道它的取值范围[a,b]，假设wi的值在实数区间[a,b]上的概率是均匀分布的。现在JIH只好去进行第二次调查，于是我们想知道每个城市建玩具厂的概率是多少。

 

 

Input

第一行有三个正整数N，a，b。

接下来N-1行每行一个正实数，为w[1]到w[N-1]。

接下来N行每行一个正实数，为d[1]到d[N]。

 

Output

N行，每行一个实数，表示在第i个城市建厂的概率

 

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

HINT

当w[5]<18.75时，将在1或4号城市建玩具厂，当w[5]>18.75时，将在5号城市建玩具厂，

当w[5]=18.75时，将在1或4或5建玩具厂。

100%的数据中：N<=100000,a<=b<=10000,w[i]<=10000,d[i]<=10。

 

 

先将每个点的代价表示为kx+b的形式，然后维护个凸包求出每段区间最低的直线，进而求出每条直线即每个点被选的概率。

现在瓶颈在求出每个点的b。

如果将厂设在i-1点 l[j],r[j]为i-1点和j点的两条路长度，则

b[i-1]=Σl[j]*r[j]*w[j]

同理b[i]=Σ(l[j]+d[i])*(r[j]-d[i])*w[j]

作差得 b[i]-b[i-1]=Σw[j]*(r[j]*d[i]-l[j]*d[i]-d[i]*d[i])

=Σw[j]*r[j]*d[i]-Σw[j]*l[j]*d[i]-Σw[j]*d[i]*d[i]

维护Σw[j]*r[j]和Σw[j]*l[j]即可。

先暴力求出第1个点的b，然后O(1)转移求其他点的b，求所有点的b变成了O(N)。

注意：有重复的直线，他们要平分概率。

code：

 #include<cstdio>
 #include<iostream>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #define maxn 100005
 #define eps 1E-9
 #define inf 1E100
 using namespace std;
 int n,flag,cnt,siz,pos[maxn];
 double w[maxn],d[maxn],dist[maxn],tot,tmp,ans,t1,t2,t3,val[maxn];
 double a,b,l,r,f[maxn],p[maxn];
 struct Line{
     double k,b;
     int id;
 }line[maxn],tl[maxn];
 bool cmp(Line x,Line y){
     if (abs(x.k-y.k)>eps) return x.k>y.k;
     return x.b<y.b;
 }
 double calc(Line l1,Line l2){return -(l1.b-l2.b)/(l1.k-l2.k);}
 struct DATA{
     double x;
     int id;
 }list[maxn];
 void solve(){
     memcpy(tl,line,sizeof(line));
     sort(tl+,tl+n+,cmp);
     for (int i=,j,t;i<=n;i=j){
         line[++cnt]=tl[i],t=;
         for (j=i;tl[j].k==tl[i].k;j++) if (tl[j].b==tl[i].b) pos[tl[j].id]=cnt,t++;
         for (j=i;tl[j].k==tl[i].k;j++) if (tl[j].b==tl[i].b) p[tl[j].id]=1.0/t;
     }
     list[siz=]=(DATA){-inf,};
     for (int i=;i<=cnt;i++){
         double t=calc(line[i],line[list[siz].id]);
         while (t<list[siz].x) siz--,t=calc(line[i],line[list[siz].id]);
         list[++siz]=(DATA){t,i};
     }
     list[++siz]=(DATA){inf,cnt+};
     l=a,r=b,flag=;
     for (int i=;i<=siz&&flag!=;i++){
         if (list[i].x>l){
             if (!flag) flag=,f[list[i-].id]=(list[i].x-l)/(r-l);
             else if (flag==&&list[i].x<r) f[list[i-].id]=(list[i].x-list[i-].x)/(r-l);
             else if (list[i].x>=r) f[list[i-].id]=(r-list[i-].x)/(r-l),flag=;
         }
     }
     for (int i=;i<=n;i++) p[i]=f[pos[i]]*p[i];
     for (int i=;i<=n;i++) printf("%.3lf\n",p[i]);
 }
 int main(){
     scanf("%d%lf%lf",&n,&a,&b);
     for (int i=;i<n;i++) scanf("%lf",&w[i]);
     for (int i=;i<=n;i++) scanf("%lf",&d[i]);
     for (int i=;i<=n;i++) dist[i]=d[i];
     for (int i=;i<=n;i++) dist[i]+=dist[i-];
     tot=dist[n];
     if (a==b){
         w[n]=a;
         t1=w[]*tot,t3=w[];
         for (int i=;i<=n;i++){
             tmp+=w[i]*(dist[i]-dist[])*(tot-dist[i]+dist[]);
             t1+=w[i]*(dist[i]-dist[]);
             t2+=w[i]*(tot-dist[i]+dist[]);
             t3+=w[i];
         }
         val[]=ans=tmp;
         for (int i=;i<=n;i++){
             tmp+=t1*d[i]-t2*d[i]-t3*d[i]*d[i];
             ans=min(ans,tmp);
             val[i]=tmp;
             t1+=w[i]*tot,t2-=w[i]*tot;
             t1-=t3*d[i],t2+=t3*d[i];
         }
         cnt=;
         for (int i=;i<=n;i++) if (abs(val[i]-ans)<=eps) cnt++;
         for (int i=;i<=n;i++) if (abs(val[i]-ans)<=eps) printf("%.3lf\n",1.0/cnt);
         else puts("0.000");
         return ;
     }
     l=,r=tot;
     for (int i=;i<=n;i++) l+=d[i],r-=d[i],line[i].k=l*r;
     t1=w[]*tot,t3=w[];
     for (int i=;i<=n;i++){
         tmp+=w[i]*(dist[i]-dist[])*(tot-dist[i]+dist[]);
         t1+=w[i]*(dist[i]-dist[]);
         t2+=w[i]*(tot-dist[i]+dist[]);
         t3+=w[i];
     }
     line[].b=tmp;
     for (int i=;i<=n;i++){
         tmp+=t1*d[i]-t2*d[i]-t3*d[i]*d[i];
         line[i].b=tmp;
         t1+=w[i]*tot,t2-=w[i]*tot;
         t1-=t3*d[i],t2+=t3*d[i];
     }
     for (int i=;i<=n;i++) line[i].id=i;
     solve();
     return ;
 }