predict.go

package main

import "fmt"
import "os"
import "bufio"
import "time"

func main(){
    now := time.Now()
    input,_ := os.Open("testing/test_cases.txt")
    defer input.Close()
    output,_ := os.Create("testing/Goutput.txt")
    defer output.Close()


    var cases int
    var gain,p float64
    fmt.Fscanf(input,"%d",&cases)

    for ;cases > 0;cases--{
        gain = 0.0
        fmt.Fscanf(input,"%f",&p)
        if p >= 0.5{
            gain = 10000 * (1-p) * (2*p -1)
        }else{
            gain = p*(1-2*p)*10000
        }
        fmt.Fprintln(output,gain+10000)
    }
    fmt.Println("Took ",time.Since(now),"seconds")
}