結果

問題 No.1629 Sorting Integers (SUM of M)
ユーザー ccppjsrb
提出日時 2021-07-30 21:22:50
言語 Go
(1.23.4)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 4,634 bytes
コンパイル時間 16,458 ms
コンパイル使用メモリ 235,428 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-15 22:49:35
合計ジャッジ時間 14,839 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

package main
import (
    "bufio"
    "fmt"
    "io"
    "os"
    "strconv"
)
var iost *Iost
type Iost struct {
    Scanner *bufio.Scanner
    Writer  *bufio.Writer
}
func NewIost(fp io.Reader, wfp io.Writer) *Iost {
    const BufSize = 2000005
    scanner := bufio.NewScanner(fp)
    scanner.Split(bufio.ScanWords)
    scanner.Buffer(make([]byte, BufSize), BufSize)
    return &Iost{Scanner: scanner, Writer: bufio.NewWriter(wfp)}
}
func (i *Iost) Text() string {
    if !i.Scanner.Scan() {
        panic("scan failed")
    }
    return i.Scanner.Text()
}
func (i *Iost) Atoi(s string) int                 { x, _ := strconv.Atoi(s); return x }
func (i *Iost) GetNextInt() int                   { return i.Atoi(i.Text()) }
func (i *Iost) Atoi64(s string) int64             { x, _ := strconv.ParseInt(s, 10, 64); return x }
func (i *Iost) GetNextInt64() int64               { return i.Atoi64(i.Text()) }
func (i *Iost) Atof64(s string) float64           { x, _ := strconv.ParseFloat(s, 64); return x }
func (i *Iost) GetNextFloat64() float64           { return i.Atof64(i.Text()) }
func (i *Iost) Print(x ...interface{})            { fmt.Fprint(i.Writer, x...) }
func (i *Iost) Printf(s string, x ...interface{}) { fmt.Fprintf(i.Writer, s, x...) }
func (i *Iost) Println(x ...interface{})          { fmt.Fprintln(i.Writer, x...) }
func isLocal() bool                               { return os.Getenv("I") == "IronMan" }
func main() {
    fp := os.Stdin
    wfp := os.Stdout
    if isLocal() {
        fp, _ = os.Open(os.Getenv("END_GAME"))
    }
    iost = NewIost(fp, wfp)
    defer func() {
        if isLocal() {
            iost.Println(recover())
        }
        iost.Writer.Flush()
    }()
    solve()
    for i := 0; isLocal() && i < 100; i++ {
        iost.Println("-----------------------------------")
        solve()
    }
}
func solve() {
    SetMod(Mod1000000007)
    n := iost.GetNextInt()
    cc := make([]int, 10)
    for i := 1; i < 10; i++ {
        cc[i] = iost.GetNextInt()
    }
    mc := newModcom(n)
    var ans Mint
    for i := 1; i < 10; i++ {
        if cc[i] == 0 {
            continue
        }
        x := mc.Facs[n-1].Mul(Mint(i))
        for j := 1; j < 10; j++ {
            if cc[j] == 0 {
                continue
            }
            if i == j {
                x.MulAs(mc.Invs[cc[j]-1])
                continue
            }
            x.MulAs(mc.Invs[cc[j]])
        }
        ans.AddAs(x)
    }
    y := Mint(0)
    for i := 0; i < n; i++ {
        y.MulAs(10)
        y.AddAs(1)
    }
    iost.Println(ans.Mul(y))
}
// Mod constants.
const (
    Mod1000000007 = 1000000007
    Mod998244353  = 998244353
)
var (
    mod  Mint
    fmod func(Mint) Mint
)
// Mint treats the modular arithmetic
type Mint int64
// SetMod sets the mod. It must be called first.
func SetMod(newmod Mint) {
    switch newmod {
    case Mod1000000007:
        fmod = staticMod1000000007
    case Mod998244353:
        fmod = staticMod998244353
    default:
        mod = newmod
        fmod = dynamicMod
    }
}
func dynamicMod(m Mint) Mint {
    m %= mod
    if m < 0 {
        return m + mod
    }
    return m
}
func staticMod1000000007(m Mint) Mint {
    m %= Mod1000000007
    if m < 0 {
        return m + Mod1000000007
    }
    return m
}
func staticMod998244353(m Mint) Mint {
    m %= Mod998244353
    if m < 0 {
        return m + Mod998244353
    }
    return m
}
// Mod returns m % mod.
func (m Mint) Mod() Mint {
    return fmod(m)
}
// Inv returns modular multiplicative inverse
func (m Mint) Inv() Mint {
    return m.Pow(Mint(0).Sub(2))
}
// Pow returns m^n
func (m Mint) Pow(n Mint) Mint {
    p := Mint(1)
    for n > 0 {
        if n&1 == 1 {
            p.MulAs(m)
        }
        m.MulAs(m)
        n >>= 1
    }
    return p
}
// Add returns m+x
func (m Mint) Add(x Mint) Mint {
    return (m + x).Mod()
}
// Sub returns m-x
func (m Mint) Sub(x Mint) Mint {
    return (m - x).Mod()
}
// Mul returns m*x
func (m Mint) Mul(x Mint) Mint {
    return (m * x).Mod()
}
// Div returns m/x
func (m Mint) Div(x Mint) Mint {
    return m.Mul(x.Inv())
}
// AddAs assigns *m + x to *m and returns m
func (m *Mint) AddAs(x Mint) *Mint {
    *m = m.Add(x)
    return m
}
// SubAs assigns *m - x to *m and returns m
func (m *Mint) SubAs(x Mint) *Mint {
    *m = m.Sub(x)
    return m
}
// MulAs assigns *m * x to *m and returns m
func (m *Mint) MulAs(x Mint) *Mint {
    *m = m.Mul(x)
    return m
}
// DivAs assigns *m / x to *m and returns m
func (m *Mint) DivAs(x Mint) *Mint {
    *m = m.Div(x)
    return m
}
type modcom struct {
    Facs, Invs []Mint
}
func newModcom(n int) modcom {
    facs := make([]Mint, n+1)
    invs := make([]Mint, n+1)
    facs[0] = 1
    for i := 0; i < n; i++ {
        facs[i+1] = facs[i].Mul(Mint(i + 1))
    }
    invs[n] = facs[n].Inv()
    for i := n - 1; i >= 0; i-- {
        invs[i] = invs[i+1].Mul(Mint(i + 1))
    }
    return modcom{
        Facs: facs,
        Invs: invs,
    }
}
func (mc *modcom) Choose(n, k int) Mint {
    if n < len(mc.Facs) {
        return mc.Facs[n].Mul(mc.Invs[k]).Mul(mc.Invs[n-k])
    }
    ans := mc.Invs[k]
    for k > 0 {
        ans.MulAs(Mint(n).Mod())
        n--
        k--
    }
    return ans
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0