DouKaku?beta

challenge 自然数の分割 

自然数nとm(n>=m>0)が与えられたとき，nをm個の非負の整数の和で表すやり方を全て出力してください．
その際，和の組(x_1, ..., x_m)は大きい順に出力してください．
ここでm = 3の時の「(a, b, c)が(A, B, C)より大きい」とは
(a > A)
(a == A) かつ (b > B)
(a == A) かつ (b == B) かつ (c > C)
のいずれかが成り立つとき(つまりは辞書的順序)とします．

例:n = 5, m = 3が与えられたときは
5, 0, 0,
4, 1, 0,
4, 0, 1,
3, 2, 0,
3, 1, 1,
...
0, 1, 4,
0, 0, 5,
を出力する．
post comment

Posted feedbacks - Scala

とりえあず。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
def sumPatterns(n:int, m:int) = {
  def sum(ns:List[int]) = (0 /: ns){_+_}
  def cmp(a:List[int],b:List[int]):boolean = {
    if(a.head == b.head) cmp(a.tail, b.tail)
    else if(a.head > b.head) true
    else false
  }

  ((List(List[int]())) /: List.make(m, 0 to n)){
    for(i <-_; j <-_) yield j::i
  }.filter(sum(_) == n).sort(cmp _)
}

Index

Feed

Other

Link

Pathtraq

loading...
Creative Commons License このサイトの内容は、 クリエイティブ・コモンズ・ライセンスの下でライセンスされています。 [詳細]