Choose random cards
This commit is contained in:
parent
48fefc813f
commit
7ec4ce70a2
32
sd.go
32
sd.go
|
@ -4,6 +4,7 @@ import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"math/rand"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
|
|
||||||
|
@ -12,11 +13,14 @@ import (
|
||||||
|
|
||||||
var file = flag.String("f", "cards", "cards file")
|
var file = flag.String("f", "cards", "cards file")
|
||||||
|
|
||||||
|
var sum = 0
|
||||||
var seg = []int{}
|
var seg = []int{}
|
||||||
|
|
||||||
func build(a *sql.Rows, l int, r int, n int) {
|
func build(a *sql.Rows, l int, r int, n int) {
|
||||||
if l == r {
|
if l == r {
|
||||||
|
a.Next()
|
||||||
a.Scan(&seg[n])
|
a.Scan(&seg[n])
|
||||||
|
sum += seg[n]
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
m := (l + r) >> 1
|
m := (l + r) >> 1
|
||||||
|
@ -27,7 +31,7 @@ func build(a *sql.Rows, l int, r int, n int) {
|
||||||
|
|
||||||
func update(x int, v int, l int, r int, n int) {
|
func update(x int, v int, l int, r int, n int) {
|
||||||
if l == r {
|
if l == r {
|
||||||
seg[n] += v
|
seg[n] = v
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
m := (l + r) >> 1
|
m := (l + r) >> 1
|
||||||
|
@ -41,7 +45,7 @@ func update(x int, v int, l int, r int, n int) {
|
||||||
|
|
||||||
func query(v int, l int, r int, n int) (int, int) {
|
func query(v int, l int, r int, n int) (int, int) {
|
||||||
if l == r {
|
if l == r {
|
||||||
return seg[n], n
|
return seg[n], l
|
||||||
}
|
}
|
||||||
m := (l + r) >> 1
|
m := (l + r) >> 1
|
||||||
if seg[n<<1] < v {
|
if seg[n<<1] < v {
|
||||||
|
@ -65,7 +69,7 @@ func main() {
|
||||||
seg = make([]int, 4*N)
|
seg = make([]int, 4*N)
|
||||||
|
|
||||||
// Build segment tree
|
// Build segment tree
|
||||||
rows, err := db.Query("SELECT * FROM cards")
|
rows, err := db.Query("SELECT weight FROM cards")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
@ -80,16 +84,30 @@ func main() {
|
||||||
defer exec.Command("stty", "-F", "/dev/tty", "echo").Run()
|
defer exec.Command("stty", "-F", "/dev/tty", "echo").Run()
|
||||||
|
|
||||||
for {
|
for {
|
||||||
fmt.Println("hello world")
|
// Choose a random card
|
||||||
|
x := rand.Intn(sum)
|
||||||
|
fmt.Println(x)
|
||||||
|
w, i := query(x, 0, N-1, 1)
|
||||||
|
fmt.Println(w)
|
||||||
|
fmt.Println(i)
|
||||||
|
|
||||||
|
var key, val string
|
||||||
|
db.QueryRow("SELECT * FROM cards WHERE idx=?", i).Scan(&i, &w, &key, &val)
|
||||||
|
fmt.Println(key)
|
||||||
|
fmt.Println(val)
|
||||||
|
|
||||||
var b []byte = make([]byte, 1)
|
var b []byte = make([]byte, 1)
|
||||||
os.Stdin.Read(b)
|
os.Stdin.Read(b)
|
||||||
if b[0] == byte('y') {
|
if b[0] == byte('y') {
|
||||||
|
w >>= 1
|
||||||
|
sum -= w
|
||||||
} else if b[0] == byte('n') {
|
} else if b[0] == byte('n') {
|
||||||
|
sum += w
|
||||||
|
w <<= 1
|
||||||
} else {
|
} else {
|
||||||
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
update(i, w, 0, N-1, 1)
|
||||||
|
db.Query("UPDATE cards SET weight=? WHERE idx=?", w, i)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Reference in a new issue