AtCoderのABC
次のABCの問題へ
前のABCの問題へ
ABC247-D Cylinder
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("4");
WillReturn.Add("1 2 3");
WillReturn.Add("2 2");
WillReturn.Add("1 3 4");
WillReturn.Add("2 3");
//4
//8
}
else if (InputPattern == "Input2") {
WillReturn.Add("2");
WillReturn.Add("1 1000000000 1000000000");
WillReturn.Add("2 1000000000");
//1000000000000000000
}
else if (InputPattern == "Input3") {
WillReturn.Add("5");
WillReturn.Add("1 1 1");
WillReturn.Add("1 1 1");
WillReturn.Add("1 1 1");
WillReturn.Add("1 1 1");
WillReturn.Add("1 1 1");
}
else {
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
}
return WillReturn;
}
class ItemInfo
{
internal long Val;
internal long Cnt;
}
static void Main()
{
List<string> InputList = GetInputList();
long[] wkArr = { };
Action<string> SplitAct = pStr =>
wkArr = pStr.Split(' ').Select(pX => long.Parse(pX)).ToArray();
var sb = new System.Text.StringBuilder();
var InsLinkedList = new LinkedList<ItemInfo>();
foreach (string EachStr in InputList.Skip(1)) {
SplitAct(EachStr);
long Type = wkArr[0];
if (Type == 1) {
long X = wkArr[1];
long C = wkArr[2];
var WillAdd = new ItemInfo();
WillAdd.Val = X;
WillAdd.Cnt = C;
InsLinkedList.AddLast(WillAdd);
}
if (Type == 2) {
long RestCnt = wkArr[1];
long Sum = 0;
while (RestCnt > 0) {
ItemInfo FirstItem = InsLinkedList.First();
long TakeCnt = Math.Min(FirstItem.Cnt, RestCnt);
FirstItem.Cnt -= TakeCnt;
Sum += FirstItem.Val * TakeCnt;
RestCnt -= TakeCnt;
if (FirstItem.Cnt == 0) {
InsLinkedList.RemoveFirst();
}
}
sb.AppendLine(Sum.ToString());
}
}
Console.Write(sb.ToString());
}
}
解説
ランレングス圧縮して、LinkedListを使ってシュミレーションしてます。
LinkedListの代わりに、Queueを使っても良いです。