Submission #1695287

---

Source Code Expand

// use std::ops::{Index, IndexMut};
// use std::cmp::{Ordering, min, max};
// use std::collections::{BinaryHeap, BTreeMap};
// use std::collections::btree_map::Entry::{Occupied, Vacant};
// use std::clone::Clone;

fn getline() -> String{
    let mut res = String::new();
    std::io::stdin().read_line(&mut res).ok();
    res
}

macro_rules! readl {
    ($t: ty) => {
        {
            let s = getline();
            s.trim().parse::<$t>().unwrap()
        }
    };
    ($( $t: ty),+ ) => {
        {
            let s = getline();
            let mut iter = s.trim().split(' ');
            ($(iter.next().unwrap().parse::<$t>().unwrap(),)*) 
        }
    };
}

macro_rules! readlvec {
    ($t: ty) => {
        {
            let s = getline();
            let iter = s.trim().split(' ');
            iter.map(|x| x.parse().unwrap()).collect::<Vec<$t>>()
        }
    }
}

macro_rules! mvec {
    ($v: expr, $s: expr) => {
        vec![$v; $s]
    };
    ($v: expr, $s: expr, $($t: expr),*) => {
        vec![mvec!($v, $($t),*); $s]
    };
}

macro_rules! debug {
    ($x: expr) => {
        println!("{}: {:?}", stringify!($x), $x)
    }
}

fn printiter<'a, T>(v: &'a T)
where
    &'a T: std::iter::IntoIterator, 
    <&'a T as std::iter::IntoIterator>::Item: std::fmt::Display {
    for (i,e) in v.into_iter().enumerate() {
        if i != 0 {
            print!(" ");
        }
        print!("{}", e);
    }
    println!("");
}

fn main() {
    let n = readl!(usize);
    let a = readlvec!(i64);
    let x = (n as i64)*(n as i64+1)/2;
    let sum: i64 = a.iter().sum();
    if sum%x != 0 {
        println!("NO");
    } else {
        let num = sum/x;
        // debug!(num);
        let mut ans = "YES";
        let ad = {
            let mut res = Vec::with_capacity(n);
            for i in 0..n-1 {
                res.push(a[i+1]-a[i]);
            }
            res.push(a[0]-a[n-1]);
            res = res.into_iter().map(|x| x-num).collect();
            res
        };
        // debug!(ad);
        for e in &ad {
            if !(*e <= 0 && *e%n as i64 == 0) {
                ans = "NO";
            }
        }
        println!("{}", ans);
    }
    
}

Submission Info

Compile Error

warning: function is never used: `printiter`, #[warn(dead_code)] on by default
  --> ./Main.rs:54:1
   |
54 | fn printiter<'a, T>(v: &'a T)
   | ^

Judge Result