ABC477-D Masking Tape

問題へのリンク

C#のソース

using System;
using System.Collections.Generic;
using System.Linq;

class Program
{
    static string InputPattern = "InputX";

    static List<string> GetInputList()
    {
        var WillReturn = new List<string>();

        if (InputPattern == "Input1") {
            WillReturn.Add("3 5");
            WillReturn.Add("1 2");
            WillReturn.Add("2 b");
            WillReturn.Add("1 1");
            WillReturn.Add("1 2");
            WillReturn.Add("2 c");
            //bcc
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("10 15");
            WillReturn.Add("1 8");
            WillReturn.Add("2 m");
            WillReturn.Add("1 3");
            WillReturn.Add("1 10");
            WillReturn.Add("2 q");
            WillReturn.Add("1 6");
            WillReturn.Add("1 10");
            WillReturn.Add("2 d");
            WillReturn.Add("1 1");
            WillReturn.Add("2 z");
            WillReturn.Add("1 9");
            WillReturn.Add("2 f");
            WillReturn.Add("1 4");
            WillReturn.Add("1 7");
            WillReturn.Add("2 k");
            //dkmfkqfazk
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static long[] GetSplitArr(string pStr)
    {
        return (pStr == "" ? new string[0] : pStr.Split(' ')).Select(pX => long.Parse(pX)).ToArray();
    }

    struct QueryInfoDef
    {
        internal long Type;
        internal long Type1Pos;
        internal char Type2Color;
    }
    static List<QueryInfoDef> mQueryInfoList = new List<QueryInfoDef>();

    static void Main()
    {
        List<string> InputList = GetInputList();
        long[] wkArr = GetSplitArr(InputList[0]);
        long N = wkArr[0];

        // 決定した色[Pos]なDict
        var AnswerDict = new Dictionary<long, char>();

        // 塗れるPosのSet
        var CanPaintPosSet = new HashSet<long>();
        for (long I = 1; I <= N; I++) {
            CanPaintPosSet.Add(I);
        }

        foreach (string EachStr in InputList.Skip(1)) {
            string[] SplitArr = EachStr.Split(' ');
            QueryInfoDef WillAdd;
            if (SplitArr[0] == "1") {
                WillAdd.Type = 1;
                WillAdd.Type1Pos = long.Parse(SplitArr[1]);
                WillAdd.Type2Color = '\0';
            }
            else {
                WillAdd.Type = 2;
                WillAdd.Type1Pos = -1;
                WillAdd.Type2Color = SplitArr[1][0];
            }
            mQueryInfoList.Add(WillAdd);
        }

        foreach (QueryInfoDef EachQueryInfo in mQueryInfoList) {
            if (EachQueryInfo.Type == 1) {
                long TargetPos = EachQueryInfo.Type1Pos;
                if (CanPaintPosSet.Contains(TargetPos)) {
                    CanPaintPosSet.Remove(TargetPos);
                }
                else {
                    CanPaintPosSet.Add(TargetPos);
                }
            }
        }

        for (int I = mQueryInfoList.Count - 1; 0 <= I; I--) {
            QueryInfoDef EachQueryInfo = mQueryInfoList[I];

            long CurrType = EachQueryInfo.Type;
            long CurrType1Pos = EachQueryInfo.Type1Pos;
            char CurrType2Color = EachQueryInfo.Type2Color;

            if (CurrType == 1) {
                if (CanPaintPosSet.Contains(CurrType1Pos)) {
                    CanPaintPosSet.Remove(CurrType1Pos);
                }
                else {
                    if (AnswerDict.ContainsKey(CurrType1Pos) == false) {
                        CanPaintPosSet.Add(CurrType1Pos);
                    }
                }
            }
            if (CurrType == 2) {
                var WillRemoveSet = new HashSet<long>();
                foreach (long EachPos in CanPaintPosSet) {
                    //Console.WriteLine("{0}を{1}で塗ります", EachPos, CurrType2Color);
                    AnswerDict[EachPos] = CurrType2Color;
                    WillRemoveSet.Add(EachPos);
                }
                CanPaintPosSet.ExceptWith(WillRemoveSet);
                CanPaintPosSet.TrimExcess();
            }
        }

        var sb = new System.Text.StringBuilder();
        for (long I = 1; I <= N; I++) {
            if (AnswerDict.ContainsKey(I)) {
                sb.Append(AnswerDict[I]);
            }
            else {
                sb.Append('a');
            }
        }
        Console.WriteLine(sb.ToString());
    }
}

解説

色塗りなので、逆から操作します。
塗れるPosのSetと、決定した色[Pos]なDict
を持って、逆から操作して解くことができます。

最悪計算量なケースで
HashSetに30万件登録してからRemoveを行うので
TrimExcessメソッドを使わないとTLEします。
TrimExcessメソッドを使わないとO(NQ)でO(30万 * 30万)になるからです。

2026-09-27 コンテストでのメモ
TrimExcessメソッドを知らなかったのでTLEが11回
AVL木で代用するか、TrimExcessメソッドを知ってればACできたはず