Merge pull request #1446 from AtCoder-NoviSteps/#1445

✨ Enable to import contests for Trie tree (#1445)

Author: KATO-Hiro

src/lib/types/contest.ts

@@ -47,3 +47,19 @@ export const ContestType: { [key in ContestTypeOrigin]: key } = {
 
 // Re-exporting the original type with the original name.
 export type ContestType = ContestTypeOrigin;
+
+/**
+ * Represents a mapping of contest IDs to contest names.
+ *
+ * @interface ContestPrefix
+ * @property {string} [key] - The contest ID.
+ * @property {string} [key: string] - The contest name associated with the contest ID.
+ * @example
+ * {
+ *   "abc001": "AtCoder Beginner Contest 001",
+ *   "arc123": "AtCoder Regular Contest 123"
+ * }
+ */
+export interface ContestPrefix {
+  [key: string]: string;
+}

src/lib/utils/contest.ts

@@ -1,4 +1,4 @@
-import { ContestType } from '$lib/types/contest';
+import { ContestType, type ContestPrefix } from '$lib/types/contest';
 
 // See:
 // https://github.com/kenkoooo/AtCoderProblems/blob/master/atcoder-problems-frontend/src/utils/ContestClassifier.ts
@@ -55,36 +55,15 @@ export const classifyContest = (contest_id: string) => {
     return ContestType.MATH_AND_ALGORITHM;
   }
 
-  // HACK: 2024年10月上旬時点では、以下のコンテストが該当。
-  // Note: 対象コンテストが増えた場合は、判定条件を見直す必要がある。
-  if (contest_id === 'tenka1-2018') {
+  if (arcLikePrefixes.has(contest_id)) {
     return ContestType.ARC_LIKE;
   }
 
-  // ・CODE FESTIVAL 2017 qual
-  // ・CODE FESTIVAL 2017 Final
-  const prefixForAgcLike = ['code-festival-2017-qual', 'cf17-final'];
-
-  if (prefixForAgcLike.some((prefix) => contest_id.startsWith(prefix))) {
+  if (agcLikePrefixes.some((prefix) => contest_id.startsWith(prefix))) {
     return ContestType.AGC_LIKE;
   }
 
-  // ・Chokudai SpeedRun
-  // ・CODE FESTIVAL 2014 決勝
-  // ・Donutsプロコンチャレンジ
-  // ・MUJIN Programming Challenge 2016
-  // ・COLOCON
-  // ・GigaCode
-  const prefixForOthers = [
-    'chokudai_S',
-    'code-festival-2014-final',
-    'donuts',
-    'mujin-pc-2016',
-    'colopl',
-    'gigacode',
-  ];
-
-  if (prefixForOthers.some((prefix) => contest_id.startsWith(prefix))) {
+  if (atCoderOthersPrefixes.some((prefix) => contest_id.startsWith(prefix))) {
     return ContestType.OTHERS;
   }
 
@@ -99,18 +78,52 @@ export const classifyContest = (contest_id: string) => {
   return null;
 };
 
-export const AOJ_COURSES = {
+// HACK: As of early November 2024, the following contests are applicable.
+// Note: The classification logic may need to be revised when new contests are added.
+const ARC_LIKE: ContestPrefix = {
+  'tenka1-2018': 'Tenka1 Programmer Contest 2018',
+} as const;
+const arcLikePrefixes = new Set(getContestPrefixes(ARC_LIKE));
+
+const AGC_LIKE: ContestPrefix = {
+  'code-festival-2016-qual': 'CODE FESTIVAL 2016 qual',
+  'code-festival-2017-qual': 'CODE FESTIVAL 2017 qual',
+  'cf17-final': 'CODE FESTIVAL 2017 final',
+} as const;
+const agcLikePrefixes = getContestPrefixes(AGC_LIKE);
+
+const ATCODER_OTHERS: ContestPrefix = {
+  chokudai_S: 'Chokudai SpeedRun',
+  'code-festival-2014-final': 'Code Festival 2014 決勝',
+  donuts: 'Donutsプロコンチャレンジ',
+  'mujin-pc-2016': 'Mujin Programming Challenge 2016',
+  'tenka1-2016-final': '天下一プログラマーコンテスト2016本戦',
+  colopl: 'COLOCON',
+  gigacode: 'GigaCode',
+} as const;
+const atCoderOthersPrefixes = getContestPrefixes(ATCODER_OTHERS);
+
+// AIZU ONLINE JUDGE AOJ Courses
+export const AOJ_COURSES: ContestPrefix = {
   ITP1: 'プログラミング入門',
   ALDS1: 'アルゴリズムとデータ構造入門',
   ITP2: 'プログラミング応用',
   DPL: '組み合わせ最適化',
 } as const;
 
+export function getPrefixForAojCourses() {
+  return getContestPrefixes(AOJ_COURSES);
+}
+
 const aojCoursePrefixes = new Set(getPrefixForAojCourses()); // For O(1) lookups
 
-// AIZU ONLINE JUDGE AOJ Courses
-export function getPrefixForAojCourses() {
-  return Object.keys(AOJ_COURSES);
+/**
+ * Extracts contest prefixes (keys) from a contest prefix object.
+ * @param contestPrefixes - Object mapping contest IDs to their display names
+ * @returns Array of contest prefix strings
+ */
+export function getContestPrefixes(contestPrefixes: Record<string, string>) {
+  return Object.keys(contestPrefixes);
 }
 
 // priority: 0 (High) - 18 (Low)

src/test/lib/utils/test_cases/contest_type.ts

@@ -161,6 +161,10 @@ export const arcLike = [
 ];
 
 export const agcLike = [
+  createTestCaseForContestType('CODE FESTIVAL 2016 qual B')({
+    contestId: 'code-festival-2016-qualb',
+    expected: ContestType.AGC_LIKE,
+  }),
   createTestCaseForContestType('CODE FESTIVAL 2017 qual A')({
     contestId: 'code-festival-2017-quala',
     expected: ContestType.AGC_LIKE,
@@ -204,6 +208,10 @@ export const atCoderOthers = [
     contestId: 'mujin-pc-2016',
     expected: ContestType.OTHERS,
   }),
+  createTestCaseForContestType('天下一プログラマーコンテスト2016本戦')({
+    contestId: 'tenka1-2016-final',
+    expected: ContestType.OTHERS,
+  }),
   createTestCaseForContestType('COLOCON 2018 qual')({
     contestId: 'colopl2018-qual',
     expected: ContestType.OTHERS,