src/test/bench/shootout-ackermann.rs

   1 // Copyright 2012 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use std::os;
  12 use std::str::from_str;
  13
  14 fn ack(m: int, n: int) -> int {
  15     if m == 0 {
  16         return n + 1
  17     } else {
  18         if n == 0 {
  19             return ack(m - 1, 1);
  20         } else {
  21             return ack(m - 1, ack(m, n - 1));
  22         }
  23     }
  24 }
  25
  26 fn main() {
  27     let args = os::args();
  28     let args = if os::getenv("RUST_BENCH").is_some() {
  29         vec!("".to_string(), "12".to_string())
  30     } else if args.len() <= 1u {
  31         vec!("".to_string(), "8".to_string())
  32     } else {
  33         args.into_iter().collect()
  34     };
  35     let n = from_str::<int>(args[1].as_slice()).unwrap();
  36     println!("Ack(3,{}): {}\n", n, ack(3, n));
  37 }