結果

問題 No.404 部分門松列
ユーザー りあん
提出日時 2016-07-17 11:40:02
言語 C#(csc)
(csc 3.9.0)
結果
WA  
実行時間 -
コード長 2,719 bytes
コンパイル時間 1,322 ms
コンパイル使用メモリ 114,400 KB
実行使用メモリ 70,452 KB
最終ジャッジ日時 2024-11-06 08:12:05
合計ジャッジ時間 11,051 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 3
other WA * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 System.Text;

class Program
{
    static void Main()
    {
        var sw = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false };
        var sc = new Scan();
        int n = sc.Int;
        var a = sc.IntArr;
        var b = new int[n];
        var ar = a.ToList().Distinct().ToArray();
        Array.Sort(ar);
        int lim = ar.Length;
        var zip = new SortedDictionary<int, int>();
        for (int i = 0; i < lim; i++)
            zip.Add(ar[i], i);

        var sg = new Segtree<long>(lim, (i, j) => i + j, 0);
        for (int i = 0; i < n; i++)
        {
            b[i] = zip[a[i]];
        }
        long ans = 0;
        for (int i = 0; i < n; i++)
        {
            ans += sg.run(0, b[i]) * sg.run(b[i] + 1, lim);
            sg.update(b[i], sg.look(b[i]) + 1);
        }
        sg = new Segtree<long>(lim, (i, j) => i + j, 0);
        for (int i = n - 1; i >= 0; i--)
        {
            ans += sg.run(0, b[i]) * sg.run(b[i] + 1, lim);
            sg.update(b[i], sg.look(b[i]) + 1);
        }
        sw.WriteLine(ans);
        sw.Flush();
    }
}
class Scan
{
    public int Int { get { return int.Parse(Str); } }
    public long Long { get { return long.Parse(Str); } }
    public string Str { get { return Console.ReadLine().Trim(); } }
    public int[] IntArr { get { return Str.Split().Select(int.Parse).ToArray(); } }
}
class Segtree<T>
{
    int n;
    T[] tree;
    Func<T, T, T> f;
    T exnum;
    public Segtree(int m, Func<T, T, T> f, T ex)
    {
        this.f = f;
        this.exnum = ex;
        n = 1;
        while (n < m) n <<= 1;

        tree = new T[(n << 1) - 1];
        for (int i = 0; i < tree.Length; i++) tree[i] = ex;
    }
    public Segtree(int m, T ini, Func<T, T, T> f, T ex)
    {
        this.f = f;
        this.exnum = ex;
        n = 1;
        while (n < m) n <<= 1;

        tree = new T[(n << 1) - 1];
        for (int i = 0; i < tree.Length; i++) tree[i] = ini;
        for (int i = 0; i < m; ++i) update(i, ini);
    }
    public void update(int j, T x)
    {
        int i = j + n - 1;
        tree[i] = x;
        while (i > 0)
        {
            i = (i - 1) >> 1;
            tree[i] = f(tree[(i << 1) + 1], tree[(i << 1) + 2]);
        }
    }
    public T look(int i) { return tree[i + n - 1]; }

    // [s, t)
    public T run(int s, int t) { return query(s, t, 0, 0, n); }
    T query(int s, int t, int k, int l, int r)
    {
        if (r <= s || t <= l) return exnum;
        if (s <= l && r <= t) return tree[k];

        return f(query(s, t, (k << 1) + 1, l, (l + r) >> 1), query(s, t, (k + 1) << 1, (l + r) >> 1, r));
    }
}
0