結果

問題 No.877 Range ReLU Query
ユーザー ひばちひばち
提出日時 2019-09-06 22:58:56
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 594 ms / 2,000 ms
コード長 11,462 bytes
コンパイル時間 2,835 ms
コンパイル使用メモリ 114,688 KB
実行使用メモリ 48,280 KB
最終ジャッジ日時 2024-11-08 10:10:16
合計ジャッジ時間 10,140 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
19,584 KB
testcase_01 AC 47 ms
20,352 KB
testcase_02 AC 46 ms
20,352 KB
testcase_03 AC 48 ms
20,352 KB
testcase_04 AC 45 ms
19,968 KB
testcase_05 AC 42 ms
19,840 KB
testcase_06 AC 45 ms
20,224 KB
testcase_07 AC 45 ms
20,096 KB
testcase_08 AC 47 ms
20,480 KB
testcase_09 AC 46 ms
19,968 KB
testcase_10 AC 48 ms
20,224 KB
testcase_11 AC 564 ms
46,292 KB
testcase_12 AC 498 ms
44,416 KB
testcase_13 AC 417 ms
39,464 KB
testcase_14 AC 417 ms
40,448 KB
testcase_15 AC 594 ms
47,128 KB
testcase_16 AC 568 ms
46,264 KB
testcase_17 AC 558 ms
46,664 KB
testcase_18 AC 551 ms
46,616 KB
testcase_19 AC 495 ms
47,900 KB
testcase_20 AC 537 ms
48,280 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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.Linq;
using System.IO;
using static System.Console;
using static System.Convert;
using static System.Math;
//using static System.Globalization.CultureInfo;
using System.Text;
class Program
{
    public static bool chmin<T>(ref T num, T val) where T : IComparable<T>
    { if (num.CompareTo(val) == 1) { num =val; return true; } return false; }
    public static bool chmax<T>(ref T num, T val) where T : IComparable<T>
    { if (num.CompareTo(val) == -1) { num = val; return true; } return false; }
    public static void swap<T>(ref T v1, ref T v2)
    { var t = v2; v2 = v1; v1 = t; }
    static void Main(string[] args)
    {
        var sw = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false };
        Console.SetOut(sw);
        new Program().Solve();
        Console.Out.Flush();
    }
    void Solve()
    {
        int num, q;
        Input.Make(out num, out q);
        var seg = new SegmentTree<long>(num, 0, (a, b) => a + b);
        var ct = new SegmentTree<long>(num, 0, (a, b) => a + b);
        var index = 0;
        var ar = new Pair<int, int>[num];
        for (var i = 0; i < num; i++)
            ar[i] = new Pair<int, int>(Input.Next<int>(), i);
        Array.Sort(ar);
        ar = ar.Reverse().ToArray();
        var query = Input.create(q, _ => new int[5]);
        for(var i = 0; i < q; i++)
        {
            for (var j = 0; j < 4; j++)
                query[i][j] = Input.Next<int>();
            query[i][4] = i;
        }
        query = query.OrderByDescending(v => v[3]).ToArray();
        var res = new long[q];
        foreach (var qq in query)
        {
            int l = qq[1], r = qq[2], x = qq[3];
            while(index!=num&&x<=ar[index].v1)
            {
                seg.Update(ar[index].v2, ar[index].v1);
                ct.Update(ar[index].v2, 1);index++;
            }
            var c = ct.Query(l - 1, r);
            res[qq[4]] = seg.Query(l - 1, r) - c * x;
        }
        WriteLine(string.Join("\n", res));
    }
}

public class LazySegmentTree<T> : SegmentTree<T>
where T : IComparable<T>
{
    protected T e, min;
    protected T[] lazy;
    protected Func<T, T, T> lazyFunc;
    protected Func<T, int, T> lazyUpdateFunc;
    public LazySegmentTree(int num, T init, T e, Func<T, T, T> func, Func<T, T, T> updateFunc = null, Func<T, T, T> lazyFunc = null, Func<T, int, T> lazyUpdateFunc = null, T min = default(T))
        : base(num, init, func, updateFunc)
    {
        this.e = e;
        this.min = min;
        if (min.CompareTo(default(T)) == 0)
            this.min = init;
        this.lazyFunc = lazyFunc ?? ((a, b) => b);
        this.lazyUpdateFunc = lazyUpdateFunc ?? ((a, b) => a);
        lazy = Enumerable.Repeat(0, 2 * this.num + 1).Select(_ => e).ToArray();
    }
    protected void eval(int len, int k)
    {
        if (lazy[k].CompareTo(e) == 0) return;
        if (Right(k) < num * 2)
        {
            lazy[Left(k)] = lazyFunc(lazy[Left(k)], lazy[k]);
            lazy[Right(k)] = lazyFunc(lazy[Right(k)], lazy[k]);
        }
        item[k] = updateFunc(item[k], lazyUpdateFunc(lazy[k], len));
        lazy[k] = e;
    }
    public override void Update(int left, int right, T value)
        => Update(left, right, 0, 0, num, value);
    protected override void Update(int left, int right, int k, int l, int r, T value)
    {
        eval(r - l, k);
        if (r <= left || right <= l) return;
        if (left <= l && r <= right)
        {
            lazy[k] = lazyFunc(lazy[k], value);
            eval(r - l, k);
        }
        else
        {
            Update(left, right, Left(k), l, (l + r) >> 1, value);
            Update(left, right, Right(k), (l + r) >> 1, r, value);
            item[k] = func(item[Left(k)], item[Right(k)]);
        }
    }
    public override T Query(int left, int right)
        => Query(left, right, 0, 0, num);
    protected override T Query(int left, int right, int k, int l, int r)
    {
        if (r <= left || right <= l) return min;
        eval(r - l, k);
        if (left <= l && r <= right) return item[k];
        else
            return func(Query(left, right, Left(k), l, (l + r) >> 1), Query(left, right, Right(k), (l + r) >> 1, r));
    }
}

public class SegmentTree<T>
{
    protected readonly T[] item;
    protected readonly int num;
    protected readonly Func<T, T, T> func;
    protected readonly Func<T, T, T> updateFunc;
    protected readonly T init;

    protected int Parent(int index)
        => (index - 1) >> 1;
    protected int Left(int index)
        => (index << 1) + 1;
    protected int Right(int index)
        => (index + 1) << 1;
    public T this[int i]
    {
        get { return item[i + num - 1]; }
        set { item[i + num - 1] = value; }
    }

    public SegmentTree(int num, T init, Func<T, T, T> func, Func<T, T, T> updateFunc = null)
    {
        this.func = func;
        this.num = 1;
        this.init = init;
        this.updateFunc = updateFunc ?? ((T val1, T val2) => val2);
        while (this.num <= num)
            this.num *= 2;
        item = new T[2 * this.num - 1];
        for (var i = 0; i < 2 * this.num - 1; i++)
            item[i] = init;
    }
    public void Update(int index, T value)
    {
        index += num - 1;
        item[index] = updateFunc(item[index], value);
        while (index > 0)
        {
            index = Parent(index);
            item[index] = func(item[Left(index)], item[Right(index)]);
        }
    }
    public virtual void Update(int left, int right, T value)
        => Update(left, right, 0, 0, num, value);
    protected virtual void Update(int left, int right, int k, int l, int r, T value)
    {
        if (r <= left || right <= l) return;
        if (left <= l && r <= right) item[k] = updateFunc(item[k], value);
        else
        {
            Update(left, right, Left(k), l, (l + r) >> 1, value);
            Update(left, right, Right(k), (l + r) >> 1, r, value);
        }
    }
    public void All_Update()
    {
        for (int i = num - 2; i >= 0; i--)
            item[i] = func(item[Left(i)], item[Right(i)]);
    }
    public T Query(int index)
    {
        index += num - 1;
        var value = func(init, item[index]);
        while (index > 0)
        {
            index = Parent(index);
            value = func(value, item[index]);
        }
        return value;
    }
    //[left,right)
    public virtual T Query(int left, int right)
        => Query(left, right, 0, 0, num);
    protected virtual T Query(int left, int right, int k, int l, int r)
    {
        if (r <= left || right <= l) return init;
        if (left <= l && r <= right) return item[k];
        else
            return func(Query(left, right, Left(k), l, (l + r) >> 1), Query(left, right, Right(k), (l + r) >> 1, r));
    }
}

public class Input
{
    public static string read => ReadLine().Trim();
    public static int[] ar => read.Split(' ').Select(int.Parse).ToArray();
    public static int num => ToInt32(read);
    public static long[] arL => read.Split(' ').Select(long.Parse).ToArray();
    public static long numL => ToInt64(read);
    public static T[] create<T>(int n, Func<int, T> f)
        => Enumerable.Repeat(0, n).Select(f).ToArray();
    public static char[][] grid(int h)
        => create(h, _ => read.ToCharArray());
    public static int[] ar1D(int n)
        => create(n, _ => num);
    public static long[] arL1D(int n)
        => create(n, _ => numL);
    public static string[] strs(int n)
        => create(n, _ => read);
    public static int[][] ar2D(int n)
        => create(n, _ => ar);
    public static long[][] arL2D(int n)
        => create(n, _ => arL);
    public static List<T>[] edge<T>(int n)
        => create(n, _ => new List<T>());
    public static T GetValue<T>(string g)
    {
        var t = typeof(T);
        if (t == typeof(int))
            return (T)(object)int.Parse(g);
        if (t == typeof(long))
            return (T)(object)long.Parse(g);
        if (t == typeof(string))
            return (T)(object)g;
        if (t == typeof(char))
            return (T)(object)char.Parse(g);
        if (t == typeof(double))
            return (T)(object)double.Parse(g);
        if (t == typeof(bool))
            return (T)(object)bool.Parse(g);
        return default(T);
    }
    public static void Make<T1, T2>(out T1 v1, out T2 v2)
    {
        v1 = Next<T1>();
        v2 = Next<T2>();
    }
    public static void Make<T1, T2, T3>(out T1 v1, out T2 v2, out T3 v3)
    {
        Make(out v1, out v2);
        v3 = Next<T3>();
    }
    public static void Make<T1, T2, T3, T4>(out T1 v1, out T2 v2, out T3 v3, out T4 v4)
    {
        Make(out v1, out v2, out v3);
        v4 = Next<T4>();
    }
    public static void Make<T1, T2, T3, T4, T5>(out T1 v1, out T2 v2, out T3 v3, out T4 v4, out T5 v5)
    {
        Make(out v1, out v2, out v3, out v4);
        v5 = Next<T5>();
    }
    public static void Make<T1, T2, T3, T4, T5, T6>(out T1 v1, out T2 v2, out T3 v3, out T4 v4, out T5 v5, out T6 v6)
    {
        Make(out v1, out v2, out v3, out v4, out v5);
        v6 = Next<T6>();
    }
    private static Queue<string> sc;
    public static T Next<T>() { sc = sc ?? new Queue<string>(); if (sc.Count == 0) foreach (var item in read.Split(' ')) sc.Enqueue(item); return GetValue<T>(sc.Dequeue()); }
    public static void Next<T>(ref T val) => val = Next<T>();
    public const long Inf = (long)1e18;
    public const double eps = 1e-6;
    public const string Alfa = "abcdefghijklmnopqrstuvwxyz";
    public const int MOD = 1000000007;
}

public class Pair<T1, T2> : IComparable<Pair<T1, T2>>
{
    public T1 v1 { get; set; }
    public T2 v2 { get; set; }
    public Pair() { v1 = Input.Next<T1>(); v2 = Input.Next<T2>(); }
    public Pair(T1 v1, T2 v2)
    { this.v1 = v1; this.v2 = v2; }

    public int CompareTo(Pair<T1, T2> p)
    {
        var c = Comparer<T1>.Default.Compare(v1, p.v1);
        if (c == 0)
            c = Comparer<T2>.Default.Compare(v2, p.v2);
        return c;
    }
    public override string ToString()
        => $"{v1.ToString()} {v2.ToString()}";
    public override bool Equals(object obj)
        => this == (Pair<T1, T2>)obj;
    public override int GetHashCode()
        => v1.GetHashCode() ^ v2.GetHashCode();
    public static bool operator ==(Pair<T1, T2> p1, Pair<T1, T2> p2)
        => p1.CompareTo(p2) == 0;
    public static bool operator !=(Pair<T1, T2> p1, Pair<T1, T2> p2)
        => p1.CompareTo(p2) != 0;
    public static bool operator >(Pair<T1, T2> p1, Pair<T1, T2> p2)
        => p1.CompareTo(p2) == 1;
    public static bool operator >=(Pair<T1, T2> p1, Pair<T1, T2> p2)
        => p1.CompareTo(p2) != -1;
    public static bool operator <(Pair<T1, T2> p1, Pair<T1, T2> p2)
        => p1.CompareTo(p2) == -1;
    public static bool operator <=(Pair<T1, T2> p1, Pair<T1, T2> p2)
        => p1.CompareTo(p2) != 1;
}

public class Pair<T1, T2, T3> : Pair<T1, T2>, IComparable<Pair<T1, T2, T3>>
{
    public T3 v3 { get; set; }
    public Pair() : base() { v3 = Input.Next<T3>(); }
    public Pair(T1 v1, T2 v2, T3 v3) : base(v1, v2)
    { this.v3 = v3; }

    public int CompareTo(Pair<T1, T2, T3> p)
    {
        var c = base.CompareTo(p);
        if (c == 0)
            c = Comparer<T3>.Default.Compare(v3, p.v3);
        return c;
    }
    public override string ToString()
        => $"{base.ToString()} {v3.ToString()}";
}

0