結果

問題 No.833 かっこいい電車
コンテスト
ユーザー EmKjp
提出日時 2019-05-24 23:33:20
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 400 ms / 2,000 ms
コード長 4,830 bytes
コンパイル時間 983 ms
コンパイル使用メモリ 109,056 KB
実行使用メモリ 34,688 KB
最終ジャッジ日時 2024-07-02 03:47:48
合計ジャッジ時間 7,234 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;

partial class Solver
{
    public void Run()
    {
        var N = ni();
        var Q = ni();
        var A = ni(N);

        var connectTree = new BinaryIndexTree(N);
        var attractiveTree = new BinaryIndexTree(N);
        for (int i = 0; i < N; i++)
        {
            attractiveTree.Add(i, A[i]);
        }

        for (int i = 0; i < Q; i++)
        {
            var q = ni();
            var x = ni() - 1;
            switch (q)
            {
                case 1: // connect
                    if (connectTree.Sum(x, x) == 0)
                        connectTree.Add(x, 1);
                    break;
                case 2:
                    if (connectTree.Sum(x, x) == 1)
                        connectTree.Add(x, -1);
                    break;
                case 3:
                    attractiveTree.Add(x, 1);
                    break;
                case 4:
                    {
                        var lower = x;
                        var upper = x;
                        for (int d = (1 << 20); d > 0; d >>= 1)
                        {
                            if (lower - d >= 0 && connectTree.Sum(lower - d, x - 1) == x - (lower - d))
                            {
                                lower -= d;
                            }
                            if (upper + d < N && connectTree.Sum(upper, upper + d - 1) == (upper + d) - upper)
                            {
                                upper += d;
                            }
                        }
                        cout.WriteLine(attractiveTree.Sum(lower, upper));
                    }
                    break;
            }
        }
    }
}

public class BinaryIndexTree
{
    private long[] array;
    public BinaryIndexTree(int size)
    {
        array = new long[size];
    }
    public long Sum(long a, long b)
    { // [a,b]
        if (a != 0) return Sum(0, b) - Sum(0, a - 1);
        long val = 0;
        for (; b >= 0; b = (b & (b + 1)) - 1) val += array[b];
        return val;
    }

    public void Add(int k, long val)
    {
        for (; k < array.Length; k |= k + 1)
            array[k] += val;
    }
}

// PREWRITEN CODE BEGINS FROM HERE
partial class Solver : Scanner
{
    public static void Main(string[] args)
    {
        new Solver(Console.In, Console.Out).Run();
    }

    TextReader cin;
    TextWriter cout;

    public Solver(TextReader reader, TextWriter writer)
        : base(reader)
    {
        this.cin = reader;
        this.cout = writer;
    }
    public Solver(string input, TextWriter writer)
        : this(new StringReader(input), writer)
    {
    }

    public int ni() { return NextInt(); }
    public int[] ni(int n) { return NextIntArray(n); }
    public long nl() { return NextLong(); }
    public long[] nl(int n) { return NextLongArray(n); }
    public double nd() { return NextDouble(); }
    public string ns() { return Next(); }
}

public class Scanner
{
    private TextReader Reader;
    private Queue<String> TokenQueue = new Queue<string>();
    private CultureInfo ci = CultureInfo.InvariantCulture;

    public Scanner()
        : this(Console.In)
    {
    }

    public Scanner(TextReader reader)
    {
        this.Reader = reader;
    }

    public int NextInt() { return Int32.Parse(Next(), ci); }
    public long NextLong() { return Int64.Parse(Next(), ci); }
    public double NextDouble() { return double.Parse(Next(), ci); }
    public string[] NextArray(int size)
    {
        var array = new string[size];
        for (int i = 0; i < size; i++) array[i] = Next();
        return array;
    }
    public int[] NextIntArray(int size)
    {
        var array = new int[size];
        for (int i = 0; i < size; i++) array[i] = NextInt();
        return array;
    }

    public long[] NextLongArray(int size)
    {
        var array = new long[size];
        for (int i = 0; i < size; i++) array[i] = NextLong();
        return array;
    }

    public String Next()
    {
        if (TokenQueue.Count == 0)
        {
            if (!StockTokens()) throw new InvalidOperationException();
        }
        return TokenQueue.Dequeue();
    }

    public bool HasNext()
    {
        if (TokenQueue.Count > 0)
            return true;
        return StockTokens();
    }

    private bool StockTokens()
    {
        while (true)
        {
            var line = Reader.ReadLine();
            if (line == null) return false;
            var tokens = line.Trim().Split(" ".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
            if (tokens.Length == 0) continue;
            foreach (var token in tokens)
                TokenQueue.Enqueue(token);
            return true;
        }
    }
}
0