#656863 (Rust) No.1505 Zero-Product Ranges

• 問題: No.1505 Zero-Product Ranges
• ユーザー: Moss_Local
• 提出日時: 2021-05-14 22:12:44
• 言語: Rust (1.77.0 + proconio)
• 結果: AC
• 実行時間: 8 ms / 2,000 ms
• コード長: 5,270 bytes
• コンパイル時間: 14,812 ms
• コンパイル使用メモリ: 384,128 KB
• 実行使用メモリ: 5,528 KB
• 最終ジャッジ日時: 2024-10-02 01:31:38
• 合計ジャッジ時間: 16,708 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 KB
testcase_03	AC	6 ms; 5,248 KB
testcase_04	AC	8 ms; 5,528 KB
testcase_05	AC	3 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	3 ms; 5,248 KB
testcase_08	AC	3 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	3 ms; 5,248 KB
testcase_11	AC	3 ms; 5,248 KB
testcase_12	AC	2 ms; 5,248 KB
testcase_13	AC	5 ms; 5,248 KB
testcase_14	AC	4 ms; 5,248 KB
testcase_15	AC	7 ms; 5,324 KB
testcase_16	AC	8 ms; 5,248 KB
testcase_17	AC	7 ms; 5,248 KB
testcase_18	AC	7 ms; 5,248 KB
testcase_19	AC	1 ms; 5,248 KB
testcase_20	AC	1 ms; 5,248 KB
testcase_21	AC	1 ms; 5,248 KB
testcase_22	AC	8 ms; 5,248 KB
testcase_23	AC	7 ms; 5,248 KB
testcase_24	AC	7 ms; 5,248 KB
testcase_25	AC	7 ms; 5,248 KB
testcase_26	AC	8 ms; 5,248 KB
testcase_27	AC	7 ms; 5,248 KB
testcase_28	AC	8 ms; 5,248 KB
testcase_29	AC	8 ms; 5,248 KB
testcase_30	AC	8 ms; 5,248 KB
testcase_31	AC	7 ms; 5,248 KB
testcase_32	AC	4 ms; 5,248 KB
testcase_33	AC	4 ms; 5,248 KB
testcase_34	AC	5 ms; 5,248 KB
testcase_35	AC	4 ms; 5,248 KB
testcase_36	AC	4 ms; 5,248 KB
testcase_37	AC	4 ms; 5,248 KB
testcase_38	AC	4 ms; 5,248 KB
testcase_39	AC	4 ms; 5,248 KB
testcase_40	AC	4 ms; 5,248 KB
testcase_41	AC	4 ms; 5,248 KB
testcase_42	AC	2 ms; 5,248 KB
testcase_43	AC	1 ms; 5,248 KB
testcase_44	AC	2 ms; 5,248 KB
testcase_45	AC	2 ms; 5,248 KB
testcase_46	AC	1 ms; 5,248 KB
testcase_47	AC	2 ms; 5,248 KB
testcase_48	AC	1 ms; 5,248 KB
testcase_49	AC	1 ms; 5,248 KB
testcase_50	AC	1 ms; 5,248 KB
testcase_51	AC	2 ms; 5,248 KB

コンパイルメッセージ

warning: unnecessary parentheses around type
  --> src/main.rs:70:15
   |
70 | fn readi() -> (i64) {
   |               ^   ^
   |
   = note: `#[warn(unused_parens)]` on by default
help: remove these parentheses
   |
70 - fn readi() -> (i64) {
70 + fn readi() -> i64 {
   |

warning: variable does not need to be mutable
   --> src/main.rs:181:9
    |
181 |     let mut vec: Vec<usize> = read_vec();
    |         ----^^^
    |         |
    |         help: remove this `mut`
    |
    = note: `#[warn(unused_mut)]` on by default

warning: variable does not need to be mutable
   --> src/main.rs:193:13
    |
193 |         let mut x = data[i + 1] - data[i];
    |             ----^
    |             |
    |             help: remove this `mut`

ソースコード

// -*- coding:utf-8-unix -*-
// #![feature(map_first_last)]
#![allow(dead_code)]
#![allow(unused_imports)]
#![allow(unused_macros)]
use std::collections::*;
use std::convert::*;
use std::convert::{From, Into};
use std::fmt::Debug;
use std::fs::File;
use std::io::prelude::*;
use std::io::*;
use std::marker::Copy;
use std::mem::*;
use std::ops::Bound::*;
use std::ops::{Add, Mul, Neg, Sub};
use std::str;
use std::vec;
use std::{cmp, process::Output};
use std::{cmp::Ordering, env::consts::DLL_PREFIX};
use std::{cmp::Ordering::*, f32::consts::PI};

const INF: i64 = 1223372036854775807;
const UINF: usize = INF as usize;
const FINF: f64 = 122337203685.0;
const INF128: i128 = 1223372036854775807000000000000;
const LINF: i64 = 2147483647;
const MOD: i64 = 1000000007;
// const MOD: i64 = 998244353;
const UMOD: usize = MOD as usize;
use std::cmp::*;
use std::collections::*;
use std::io::stdin;
use std::io::stdout;
use std::io::Write;

macro_rules! p {
    ($x:expr) => {
        println!("{}", $x);
    };
}

macro_rules! d {
    ($x:expr) => {
        dbg!($x);
    };
}

macro_rules! v {
    ($x:expr) => {
        if $x == 0 {
            vec![];
        } else if $x == 1 {
            BTreeSet::new();
        } else if $x == 2 {
            BTreeMap::new();
        }
    };
}

// use str::Chars;
#[allow(dead_code)]
fn read<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

#[allow(dead_code)]
fn readi() -> (i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    iter.next().unwrap().parse::<i64>().unwrap()
}

#[allow(dead_code)]
fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    read::<String>()
        .split_whitespace()
        .map(|e| e.parse().ok().unwrap())
        .collect()
}
#[allow(dead_code)]
fn read_vec2<T: std::str::FromStr>(n: u32) -> Vec<Vec<T>> {
    (0..n).map(|_| read_vec()).collect()
}

#[allow(dead_code)]
fn readii() -> (i64, i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
    )
}

fn readff() -> (f64, f64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<f64>().unwrap(),
        iter.next().unwrap().parse::<f64>().unwrap(),
    )
}

#[allow(dead_code)]
fn readiii() -> (i64, i64, i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
    )
}
#[allow(dead_code)]
fn readuu() -> (usize, usize) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
    )
}

fn readcc() -> (char, char) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<char>().unwrap(),
        iter.next().unwrap().parse::<char>().unwrap(),
    )
}

#[allow(dead_code)]
fn readuuu() -> (usize, usize, usize) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
    )
}

#[allow(dead_code)]
fn readuuuu() -> (usize, usize, usize, usize) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
    )
}

fn readiiii() -> (i64, i64, i64, i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
    )
}

fn solve() {
    let n: usize = read();
    let mut vec: Vec<usize> = read_vec();
    let mut data = vec![];
    data.push(0);
    for i in 0..n {
        if vec[i] == 0 {
            data.push(i + 1);
        }
    }
    data.push(n + 1);
    // d!(data.clone());
    let mut res = 0 as usize;
    for i in 0..data.len() - 1 {
        let mut x = data[i + 1] - data[i];
        if x == 0 {
            continue;
        }
        res += x * (x - 1) / 2;
    }
    println!("{:?}", n * (n - 1) / 2 + n - res);
    return;
}

fn main() {
    solve();
}