結果

問題 No.990 N×Mマス計算（Kの倍数）
ユーザー りあんりあん
提出日時 2020-02-14 22:33:22
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 246 ms / 2,000 ms
コード長 10,292 bytes
コンパイル時間 4,192 ms
コンパイル使用メモリ 113,896 KB
実行使用メモリ 43,656 KB
最終ジャッジ日時 2023-08-10 02:10:26
合計ジャッジ時間 7,066 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 69 ms
21,832 KB
testcase_01 AC 73 ms
23,980 KB
testcase_02 AC 73 ms
23,840 KB
testcase_03 AC 70 ms
23,812 KB
testcase_04 AC 72 ms
21,916 KB
testcase_05 AC 70 ms
23,960 KB
testcase_06 AC 70 ms
23,872 KB
testcase_07 AC 73 ms
26,000 KB
testcase_08 AC 70 ms
21,816 KB
testcase_09 AC 69 ms
21,688 KB
testcase_10 AC 120 ms
31,772 KB
testcase_11 AC 139 ms
33,744 KB
testcase_12 AC 245 ms
38,844 KB
testcase_13 AC 117 ms
27,424 KB
testcase_14 AC 140 ms
32,324 KB
testcase_15 AC 113 ms
25,804 KB
testcase_16 AC 132 ms
33,164 KB
testcase_17 AC 104 ms
25,668 KB
testcase_18 AC 246 ms
36,992 KB
testcase_19 AC 145 ms
27,076 KB
testcase_20 AC 172 ms
43,656 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 System.Threading;
using System.Text;
using System.Text.RegularExpressions;
using System.Diagnostics;
using static util;
using P = pair<int, int>;

class Program {
    static void Main(string[] args) {
        var sw = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false };
        var solver = new Solver(sw);
        // var t = new Thread(solver.solve, 1 << 26); // 64 MB
        // t.Start();
        // t.Join();
        solver.solve();
        sw.Flush();
    }
}

class Solver {
    StreamWriter sw;
    Scan sc;
    void Prt(string a) => sw.WriteLine(a);
    void Prt<T>(IEnumerable<T> a) => Prt(string.Join(" ", a));
    void Prt(params object[] a) => Prt(string.Join(" ", a));
    public Solver(StreamWriter sw) {
        this.sw = sw;
        this.sc = new Scan();
    }

    public void solve() {
        int n, m, k;
        sc.Multi(out n, out m, out k);
        char op = '+';
        var b = new int[m];
        {
            var s = sc.StrArr;
            op = s[0][0];
            b = s.Skip(1).Select(int.Parse).ToArray();
        }
        var a = new int[n];
        for (int i = 0; i < n; i++)
        {
            a[i] = sc.Int;
        }

        if (op == '+') {
            for (int i = 0; i < n; i++)
            {
                a[i] %= k;
            }
            for (int i = 0; i < m; i++)
            {
                b[i] %= k;
            }
            var dicb = new Dictionary<int, int>();
            for (int i = 0; i < m; i++)
            {
                if (!dicb.ContainsKey(b[i])) {
                    dicb.Add(b[i], 0);
                }
                ++dicb[b[i]];
            }

            long ans = 0;
            for (int i = 0; i < n; i++)
            {
                if (dicb.ContainsKey((k - a[i]) % k)) {
                    ans += dicb[(k - a[i]) % k];
                }
            }
            Prt(ans);
        }
        else {
            var pr = new List<P>();
            {
                int l = k;
                for (int i = 2; i * i <= l; i++)
                {
                    if (l % i == 0) {
                        int c = 0;
                        while (l % i == 0) {
                            ++c;
                            l /= i;
                        }
                        pr.Add(new P(i, c));
                    }
                }
                if (l > 1) {
                    pr.Add(new P(l, 1));
                }
            }
            var yak = new List<int>();
            for (int i = 1; i * i <= k; i++)
            {
                if (k % i == 0) {
                    yak.Add(i);
                    if (k / i != i) {
                        yak.Add(k / i);
                    }
                }
            }
            yak.Sort();
            var dicb = new Dictionary<int, int>();
            var bb = new Dictionary<int, int>();
            for (int i = 0; i < m; i++)
            {
                int p = 1;
                foreach (var item in pr)
                {
                    int d = b[i];
                    for (int j = 0; j < item.v2; j++)
                    {
                        if (d % item.v1 == 0) {
                            d /= item.v1;
                            p *= item.v1;
                        }
                        else break;
                    }
                }
                if (!bb.ContainsKey(p)) {
                    bb.Add(p, 0);
                }
                ++bb[p];
                // DBG(p);
            }
            foreach (var item in bb)
            {
                int p = item.Key;
                foreach (var j in yak)
                {
                    if (j > p) break;
                    if (p % j == 0) {
                        if (!dicb.ContainsKey(j)) {
                            dicb.Add(j, 0);
                        }
                        dicb[j] += item.Value;
                    }
                }

            }

            long ans = 0;
            for (int i = 0; i < n; i++)
            {
                int p = 1;
                foreach (var item in pr)
                {
                    int d = a[i];
                    for (int j = 0; j < item.v2; j++)
                    {
                        if (d % item.v1 == 0) {
                            d /= item.v1;
                            p *= item.v1;
                        }
                        else break;
                    }
                }
                if (dicb.ContainsKey(k / p)) {
                    ans += dicb[k / p];
                }
            }
            Prt(ans);
        }



    }
}

class pair<T, U> : IComparable<pair<T, U>> {
    public T v1;
    public U v2;
    public pair() : this(default(T), default(U)) {}
    public pair(T v1, U v2) { this.v1 = v1; this.v2 = v2; }
    public int CompareTo(pair<T, U> a) {
        int c = Comparer<T>.Default.Compare(v1, a.v1);
        return c != 0 ? c : Comparer<U>.Default.Compare(v2, a.v2);
    }
    public override string ToString() => v1 + " " + v2;
    public void Deconstruct(out T a, out U b) { a = v1; b = v2; }
}
static class util {
    public static readonly int M = 1000000007;
    // public static readonly int M = 998244353;
    public static readonly long LM = 1L << 60;
    public static readonly double eps = 1e-11;
    public static void DBG(string a) => Console.Error.WriteLine(a);
    public static void DBG<T>(IEnumerable<T> a) => DBG(string.Join(" ", a));
    public static void DBG(params object[] a) => DBG(string.Join(" ", a));
    public static void Assert(params bool[] conds) {
        foreach (var cond in conds) if (!cond) throw new Exception();
    }
    public static pair<T, U> make_pair<T, U>(T v1, U v2) => new pair<T, U>(v1, v2);
    public static int CompareList<T>(IList<T> a, IList<T> b) where T : IComparable<T> {
        for (int i = 0; i < a.Count && i < b.Count; i++)
            if (a[i].CompareTo(b[i]) != 0) return a[i].CompareTo(b[i]);
        return a.Count.CompareTo(b.Count);
    }
    public static bool inside(int i, int j, int h, int w) => i >= 0 && i < h && j >= 0 && j < w;
    static readonly int[] dd = { 0, 1, 0, -1 };
    // static readonly string dstring = "RDLU";
    public static P[] adjacents(int i, int j)
        => Enumerable.Range(0, dd.Length).Select(k => new P(i + dd[k], j + dd[k ^ 1])).ToArray();
    public static P[] adjacents(int i, int j, int h, int w)
        => Enumerable.Range(0, dd.Length).Select(k => new P(i + dd[k], j + dd[k ^ 1]))
                                         .Where(p => inside(p.v1, p.v2, h, w)).ToArray();
    public static P[] adjacents(this P p) => adjacents(p.v1, p.v2);
    public static P[] adjacents(this P p, int h, int w) => adjacents(p.v1, p.v2, h, w);
    public static Dictionary<T, int> compress<T>(this IEnumerable<T> a)
        => a.Distinct().OrderBy(v => v).Select((v, i) => new { v, i }).ToDictionary(p => p.v, p => p.i);
    public static Dictionary<T, int> compress<T>(params IEnumerable<T>[] a) => compress(a.Aggregate(Enumerable.Union));
    public static T[] inv<T>(this Dictionary<T, int> dic) {
        var res = new T[dic.Count];
        foreach (var item in dic) res[item.Value] = item.Key;
        return res;
    }
    public static void swap<T>(ref T a, ref T b) where T : struct { var t = a; a = b; b = t; }
    public static void swap<T>(this IList<T> a, int i, int j) where T : struct { var t = a[i]; a[i] = a[j]; a[j] = t; }
    public static T[] copy<T>(this IList<T> a) {
        var ret = new T[a.Count];
        for (int i = 0; i < a.Count; i++) ret[i] = a[i];
        return ret;
    }
    public static void Deconstruct<T>(this IList<T> v, out T a) {
        a = v[0];
    }
    public static void Deconstruct<T>(this IList<T> v, out T a, out T b) {
        a = v[0]; b = v[1];
    }
    public static void Deconstruct<T>(this IList<T> v, out T a, out T b, out T c) {
        a = v[0]; b = v[1]; c = v[2];
    }
    public static void Deconstruct<T>(this IList<T> v, out T a, out T b, out T c, out T d) {
        a = v[0]; b = v[1]; c = v[2]; d = v[3];
    }
    public static void Deconstruct<T>(this IList<T> v, out T a, out T b, out T c, out T d, out T e) {
        a = v[0]; b = v[1]; c = v[2]; d = v[3]; e = v[4];
    }
}

class Scan {
    StreamReader sr;
    public Scan() { sr = new StreamReader(Console.OpenStandardInput()); }
    public Scan(string path) { sr = new StreamReader(path); }
    public int Int => int.Parse(Str);
    public long Long => long.Parse(Str);
    public double Double => double.Parse(Str);
    public string Str => sr.ReadLine().Trim();
    public pair<T, U> Pair<T, U>() {
        T a; U b;
        Multi(out a, out b);
        return new pair<T, U>(a, b);
    }
    public P P => Pair<int, int>();
    public int[] IntArr => StrArr.Select(int.Parse).ToArray();
    public long[] LongArr => StrArr.Select(long.Parse).ToArray();
    public double[] DoubleArr => StrArr.Select(double.Parse).ToArray();
    public string[] StrArr => Str.Split(new[]{' '}, StringSplitOptions.RemoveEmptyEntries);
    bool eq<T, U>() => typeof(T).Equals(typeof(U));
    T ct<T, U>(U a) => (T)Convert.ChangeType(a, typeof(T));
    T cv<T>(string s) => eq<T, int>()    ? ct<T, int>(int.Parse(s))
                       : eq<T, long>()   ? ct<T, long>(long.Parse(s))
                       : eq<T, double>() ? ct<T, double>(double.Parse(s))
                       : eq<T, char>()   ? ct<T, char>(s[0])
                                         : ct<T, string>(s);
    public void Multi<T>(out T a) => a = cv<T>(Str);
    public void Multi<T, U>(out T a, out U b) {
        var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]);
    }
    public void Multi<T, U, V>(out T a, out U b, out V c) {
        var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]);
    }
    public void Multi<T, U, V, W>(out T a, out U b, out V c, out W d) {
        var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); d = cv<W>(ar[3]);
    }
    public void Multi<T, U, V, W, X>(out T a, out U b, out V c, out W d, out X e) {
        var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); d = cv<W>(ar[3]); e = cv<X>(ar[4]);
    }
}
0