結果

問題 No.3436 [Cherry 8th Tune B] この夏に何が起こるかな?
コンテスト
ユーザー kakel-san
提出日時 2026-01-23 23:05:58
言語 C#
(.NET 10.0.101)
結果
AC  
実行時間 884 ms / 4,000 ms
コード長 3,818 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 10,634 ms
コンパイル使用メモリ 170,288 KB
実行使用メモリ 228,476 KB
最終ジャッジ日時 2026-01-23 23:06:48
合計ジャッジ時間 44,214 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 43
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (119 ミリ秒)。
  main -> /home/judge/data/code/bin/Release/net10.0/main.dll
  main -> /home/judge/data/code/bin/Release/net10.0/publish/

ソースコード

diff #
raw source code 

using System;
using static System.Console;
using System.Linq;
using System.Collections.Generic;
using System.Security.Cryptography;
using Microsoft.VisualBasic;

class Program
{
    static int NN => int.Parse(ReadLine());
    static long[] NList => ReadLine().Split().Select(long.Parse).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var q = NN;
        var ans = new int[q][];
        for (var u = 0; u < q; ++u)
        {
            var c = NList;
            var tv = NList;
            var tc = NList;
            var bv = NList;
            var bc = NList;
            var s = NList;
            for (var i = 0; i < tc.Length; ++i) --tc[i];
            for (var i = 0; i < bc.Length; ++i) --bc[i];
            ans[u] = Resort((int)c[0], (int)c[1], (int)c[2], c[3], tv, tc, bv, bc, s);
        }
        WriteLine(string.Join("\n", ans.Select(ai => string.Join(" ", ai))));
    }
    static int[] Resort(int n, int m, int k, long p, long[] tv, long[] tc, long[] bv, long[] bc, long[] s)
    {
        if (n > m)
        {
            var ans = Resort(m, n, k, p, bv, bc, tv, tc, s);
            return new int[] { ans[1], ans[0] };
        }
        var setlist = new Dictionary<long, List<long>>();
        for (var i = 0; i < n; ++i) if (!setlist.ContainsKey(tc[i])) setlist[tc[i]] = new List<long>();
        var list = new List<long>();
        for (var i = 0; i < m; ++i)
        {
            if (setlist.ContainsKey(bc[i])) setlist[bc[i]].Add(bv[i]);
            list.Add(bv[i]);
        }
        foreach (var kv in setlist) kv.Value.Sort();
        list.Sort();
        var ok = 2_000_000_000L;
        var ng = 0L;
        while (ok - ng > 1)
        {
            var mid = (ok + ng) / 2;
            var count = 0L;
            for (var i = 0; i < n; ++i)
            {
                count += UpperBound(mid - tv[i], list) + UpperBound(mid - tv[i] + s[tc[i]], setlist[tc[i]]) - UpperBound(mid - tv[i], setlist[tc[i]]);
            }
            if (count >= p) ok = mid;
            else ng = mid;
        }
        var setdic = new Dictionary<long, Dictionary<long, int>>();
        foreach (var kv in setlist)
        {
            setdic[kv.Key] = new Dictionary<long, int>();
            foreach (var li in kv.Value)
            {
                if (setdic[kv.Key].ContainsKey(li)) ++setdic[kv.Key][li];
                else setdic[kv.Key][li] = 1;
            }
        }
        var dic = new Dictionary<long, int>();
        for (var i = 0; i < m; ++i)
        {
            if (dic.ContainsKey(bv[i])) ++dic[bv[i]];
            else dic[bv[i]] = 1;
        }
        for (var i = 0; i < n; ++i)
        {
            var setcount = 0;
            if (setdic[tc[i]].ContainsKey(ok - tv[i] + s[tc[i]])) setcount = setdic[tc[i]][ok - tv[i] + s[tc[i]]];
            var singlecount = 0;
            if (dic.ContainsKey(ok - tv[i])) singlecount = dic[ok - tv[i]];
            if (setcount > 0)
            {
                for (var j = 0; j < m; ++j) if (tc[i] == bc[j] && tv[i] + bv[j] - s[tc[i]] == ok) return new int[] { i + 1, j + 1 };
            }
            if (singlecount > setcount)
            {
                for (var j = 0; j < m; ++j) if (tc[i] != bc[j] && tv[i] + bv[j] == ok) return new int[] { i + 1, j + 1 };
            }
        }
        return new int[] { -1, -1 };
    }
    static int UpperBound<T>(T min, IList<T> list) where T: IComparable<T>
    {
        var ng = -1;
        var ok = list.Count;
        while (ok - ng > 1)
        {
            var mid = (ng + ok) / 2;
            if (list[mid].CompareTo(min) <= 0) ng = mid;
            else ok = mid;
        }
        return ok;
    }
}
0