AtCoderの企業コンテスト
次の企業コンテストの問題へ
前の企業コンテストの問題へ
キーエンス プログラミング コンテスト 2020 C Subarray Sum
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 2 3");
//1 2 3 4
}
else if (InputPattern == "Input2") {
WillReturn.Add("5 3 100");
//50 50 50 30 70
}
else {
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
}
return WillReturn;
}
static void Main()
{
List<string> InputList = GetInputList();
int[] wkArr = InputList[0].Split(' ').Select(pX => int.Parse(pX)).ToArray();
int N = wkArr[0];
int K = wkArr[1];
int S = wkArr[2];
var AnswerList = new List<int>();
AnswerList.AddRange(Enumerable.Repeat(S, K));
int RestCnt = N - K;
if (S == 1000000000) {
AnswerList.AddRange(Enumerable.Repeat(1, RestCnt));
}
else {
AnswerList.AddRange(Enumerable.Repeat(S + 1, RestCnt));
}
int[] AnswerIntArr = AnswerList.ToArray();
string[] AnswerStrArr = Array.ConvertAll(AnswerIntArr, pX => pX.ToString());
Console.WriteLine(string.Join(" ", AnswerStrArr));
}
}
解説
Sの項をK個作ってから、
和がSになる部分列がK個を超えないように、残りの項を追加してます。