#!/usr/bin/env node
var i =0;
var p =[];

function again(){
  console.log(`p${i}: `);
  var sum = 0;
  process.stdin.on('data',data=>{
    data = data.toString('utf8');
    data = data.slice(0,-1);
    if(data.length === 0){
      console.log(p);
      for(const j of p){
        sum = sum + Number(j);
      }
      if(sum > 1){
        i = 0;
        p = [];
        console.log('概率和大于1，请重新输入');
        console.log(`p${i}:`)
      }else if(sum <1){
        i=0;
        p=[];
        console.log('概率和小于1，请重新输入');
        console.log(`p${i}`)
      }else{
        var h = 0;
        for(const e of p){
          h = h - Number(e)*Math.log(Number(e),2)
        }
        h = (parseInt(h*1000))/1000;
        console.log(`p(h) = ${h} bit`);
        process.exit();
      }
    }else{
      if(isNaN(data)){
        console.log('请输入数字')
        console.log(`p${i}:`)

        return false;
      }else if(data>1){
        console.log('请输入小于1的数');
        console.log(`p${i}:`)

        return false;
      } else if(data<0){
        console.log('请输入大于0的数');
        console.log(`p${i}:`)

        return false;
      }else{
        p.push(Number(data));
        i = i+1;
        //console.log(p);
        console.log(`p${i}: `)
        return false;

      }
    }   
  })
}
again();