past202005-open c

bouzuya · Jan 12, 2024 · 686e0ef · 686e0ef
1 parent 3fd3fa1
commit 686e0ef
Showing 1 changed file with 23 additions and 3 deletions.
diff --git a/cargo-atcoder-1.70.0/contests/past202005-open/src/bin/c.rs b/cargo-atcoder-1.70.0/contests/past202005-open/src/bin/c.rs
@@ -1,10 +1,30 @@
-use proconio::{input, marker::Usize1};
+use proconio::input;
 
 fn main() {
     input! {
+        a: usize,
+        r: usize,
         n: usize,
-        a: [Usize1; n],
     };
-    let ans = n - a.len();
+
+    if r == 1 {
+        let x = a * r;
+        if x > 10_usize.pow(9) {
+            println!("large");
+        } else {
+            println!("{}", x);
+        }
+        return;
+    }
+
+    let mut x = a;
+    for _ in 1..n {
+        x *= r;
+        if x > 10_usize.pow(9) {
+            println!("large");
+            return;
+        }
+    }
+    let ans = x;
     println!("{}", ans);
 }