codeforces 641D. Little Artem and Random Variable

pi  表示两个骰子只有  (1 2 3......i) 出现概率 

pi= max1+max2+max3+.....maxi

suma_i 表示第一个骰子只有  (1 2 3......i)  出现概率和

sumb_i 表示第两个骰子只有  (1 2 3......i)  出现概率和

qi  表示只有 (i i+1 i+2.....n)出现前面的概率

qi=1-(min1+min2+min3+......mini)

suma_i *sumb_i =pi

(1-suma_i )*(1-sumb_i )=qi

 

解一个  一元二次方程

算出 suma_i sumb_i   知道前缀和即可知道答案 

ps:好久没写 想了好久  QAQ 

#include <iostream>
#include <math.h>
#include <stdio.h>
using namespace std;

double ma[100010];
double mi[100010];

double ans1[100010];
double ans2[100010];

void solve(double a,double b,double c,double &x,double &y)
{
    x=(-b-sqrt(fabs(b*b-4*a*c)))/(2*a);
    y=(-b+sqrt(fabs(b*b-4*a*c)))/(2*a);
    if(x>y)swap(x,y);
}

int main()
{
    int n;
    scanf("%d",&n);
    double p,q;
    for(int i=1;i<=n;i++)
    {
        scanf("%lf",&p);
        ma[i]=ma[i-1]+p;
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%lf",&q);
        mi[i]=mi[i-1]+q;
    }
    for(int i=1;i<=n;i++)
    {
        solve(1.0,-mi[i]-ma[i],ma[i],ans1[i],ans2[i]);
    }
    for(int i=1;i<=n;i++)
    {
        printf("%.8lf ",ans1[i]-ans1[i-1]);
    }
    printf("\n");
    for(int i=1;i<=n;i++)
    {
        printf("%.8lf ",ans2[i]-ans2[i-1]);
    }
    return 0;
}