#260358 (Rust) No.5002 stick xor

• 問題: No.5002 stick xor
• ユーザー: 最新の錆
• 提出日時: 2018-05-26 00:06:09
• 言語: Rust (1.77.0)
• 結果: AC
• 実行時間: 16 ms / 1,000 ms
• コード長: 3,105 bytes
• コンパイル時間: 2,794 ms
• 実行使用メモリ: 1,204 KB
• スコア: 23,235
• 最終ジャッジ日時: 2018-05-26 00:06:14
• ジャッジサーバーID （参考情報）: judge6 /

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	11 ms; 1,200 KB
testcase_01	AC	11 ms; 1,200 KB
testcase_02	AC	14 ms; 1,200 KB
testcase_03	AC	11 ms; 1,204 KB
testcase_04	AC	10 ms; 1,200 KB
testcase_05	AC	16 ms; 1,200 KB
testcase_06	AC	10 ms; 1,200 KB
testcase_07	AC	13 ms; 1,204 KB
testcase_08	AC	15 ms; 1,200 KB
testcase_09	AC	14 ms; 1,200 KB
testcase_10	AC	14 ms; 1,200 KB
testcase_11	AC	10 ms; 1,200 KB
testcase_12	AC	11 ms; 1,196 KB
testcase_13	AC	11 ms; 1,200 KB
testcase_14	AC	11 ms; 1,204 KB
testcase_15	AC	11 ms; 1,200 KB
testcase_16	AC	11 ms; 1,204 KB
testcase_17	AC	11 ms; 1,192 KB
testcase_18	AC	11 ms; 1,196 KB
testcase_19	AC	10 ms; 1,200 KB
testcase_20	AC	10 ms; 1,204 KB
testcase_21	AC	11 ms; 1,192 KB
testcase_22	AC	10 ms; 1,196 KB
testcase_23	AC	10 ms; 1,196 KB
testcase_24	AC	11 ms; 1,204 KB
testcase_25	AC	11 ms; 1,200 KB
testcase_26	AC	11 ms; 1,196 KB
testcase_27	AC	11 ms; 1,196 KB
testcase_28	AC	11 ms; 1,200 KB
testcase_29	AC	10 ms; 1,200 KB
testcase_30	AC	10 ms; 1,196 KB
testcase_31	AC	11 ms; 1,196 KB

ソースコード

use std::io::{Read, stdin};

fn main() {
    let mut buf = String::new();
    stdin().read_to_string(&mut buf).unwrap();
    let mut tok = buf.split_whitespace();
    let mut get = || tok.next().unwrap();
    macro_rules! get {
        ($t:ty) => (get().parse::<$t>().unwrap());
        () => (get!(usize));
    }
    let n = get!();
    let k = get!();
    let mut ls = vec![];
    for _ in 0..k {
        ls.push(get!());
    }
    let mut grid = vec![];
    for _ in 0..n {
        let tmp: Vec<_> =
            get().as_bytes()
                 .iter()
                 .map(|&ch| (ch - b'0') as i32)
                 .collect();
        grid.push(tmp);
    }
    // println!("{:?}", grid);
    let mut ans = vec![];
    for &l in ls.iter() {
        let mut max_sum_i = 0;
        let mut max_sum_j = 0;
        let mut max_i = 0;
        let mut max_j = 0;
        for i in 0..n {
            let mut sum = 0;
            for j in 0..l {
                sum += grid[i][j];
            }
            if sum > max_sum_i {
                max_sum_i = sum;
                max_i = i;
            }
            let mut sum = 0;
            for j in 0..l {
                sum += grid[j][i];
            }
            if sum > max_sum_j {
                max_sum_j = sum;
                max_j = i;
            }
        }
        if max_sum_i > max_sum_j {
            for j in 0..l {
                grid[max_i][j] = 1 - grid[max_i][j];
            }
            ans.push((true, max_i, l, 0));
        } else {
            for i in 0..l {
                grid[i][max_j] = 1 - grid[i][max_j];
            }
            ans.push((false, max_j, l, 0));
        }
    }
    for _ in 0..50 {
        for i in 0..ans.len() {
            let (b, j, l, p0) = ans[i];
            let mut table = vec![0; n + 1];
            let mut max_sum = 0;
            let mut max_p = 0;
            if b {
                for p in p0..p0+l {
                    grid[j][p] = 1 - grid[j][p];
                }
                for p in 0..n {
                    table[p+1] = table[p] + grid[j][p];
                }
            } else {
                for p in p0..p0+l {
                    grid[p][j] = 1 - grid[p][j];
                }
                for p in 0..n {
                    table[p+1] = table[p] + grid[p][j];
                }
            }
            for p in 0..n-l+1 {
                let sum = table[p+l] - table[p];
                if sum > max_sum {
                    max_sum = sum;
                    max_p = p;
                }
            }
            if b {
                for p in 0..l {
                    grid[j][max_p + p] = 1 - grid[j][max_p + p];
                }
            } else {
                for p in 0..l {
                    grid[max_p + p][j] = 1 - grid[max_p + p][j];
                }
            }
            ans[i] = (b, j, l, max_p);
        }
    }
    for &(b, i, l, p) in ans.iter() {
        if b {
            println!("{0} {1} {0} {2}", i + 1, p + 1, p + l);
        } else {
            println!("{1} {0} {2} {0}", i + 1, p + 1, p + l);
        }
    }
}