AtCoderのABC
次のABCの問題へ
前のABCの問題へ
ABC355-D Intersecting Intervals
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");
WillReturn.Add("1 5");
WillReturn.Add("7 8");
WillReturn.Add("3 7");
//2
}
else if (InputPattern == "Input2") {
WillReturn.Add("3");
WillReturn.Add("3 4");
WillReturn.Add("2 5");
WillReturn.Add("1 6");
//3
}
else if (InputPattern == "Input3") {
WillReturn.Add("2");
WillReturn.Add("1 2");
WillReturn.Add("3 4");
//0
}
else {
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
}
return WillReturn;
}
struct EventInfoDef
{
internal long Pos;
internal bool IsAdd;
}
static List<EventInfoDef> mEventInfoList = new List<EventInfoDef>();
static void Main()
{
List<string> InputList = GetInputList();
long[] wkArr = { };
Action<string> SplitAct = pStr =>
wkArr = pStr.Split(' ').Select(pX => long.Parse(pX)).ToArray();
foreach (string EachStr in InputList.Skip(1)) {
SplitAct(EachStr);
EventInfoDef WillAdd1;
WillAdd1.Pos = wkArr[0];
WillAdd1.IsAdd = true;
mEventInfoList.Add(WillAdd1);
EventInfoDef WillAdd2;
WillAdd2.Pos = wkArr[1];
WillAdd2.IsAdd = false;
mEventInfoList.Add(WillAdd2);
}
mEventInfoList = mEventInfoList.OrderBy(pX => pX.Pos).ThenByDescending(pX => pX.IsAdd).ToList();
long Answer = 0;
long CurrRangeCnt = 0;
foreach (EventInfoDef EachEventInfo in mEventInfoList) {
if (EachEventInfo.IsAdd) {
Answer += CurrRangeCnt;
CurrRangeCnt++;
}
else {
CurrRangeCnt--;
}
}
Console.WriteLine(Answer);
}
}
解説
座標の昇順(座標が同じなら、追加が先)で、イベントソートを行い、
区間の追加のタイミングで、ペアの件数を解に計上してます。