@xcpcio/core 0.22.0 → 0.24.0

package/src/rank.ts

@@ -7,6 +7,7 @@ import type { Submissions } from "./submission";
 import { Submission } from "./submission";
 import { TeamProblemStatistics } from "./problem";
 import { RankStatistics } from "./rank-statistics";
+import { Balloon, type Balloons } from "./balloon";
 
 export interface SelectOptionItem {
   value: string;
@@ -126,6 +127,8 @@ export class Rank {
 
   options: RankOptions;
 
+  balloons: Balloons;
+
   constructor(contest: Contest, teams: Teams, submissions: Submissions) {
     this.contest = contest;
 
@@ -142,40 +145,46 @@ export class Rank {
     this.rankStatistics = new RankStatistics();
 
     this.options = new RankOptions();
+
+    this.balloons = [];
   }
 
-  buildRank() {
+  cleanRank() {
     (() => {
-      (() => {
-        this.teams = [];
-
-        for (const [_k, v] of this.teamsMap) {
-          if (this.filterTeamByOrg(v)) {
-            continue;
-          }
+      this.teams = [];
 
-          this.teams.push(v);
+      for (const [_k, v] of this.teamsMap) {
+        if (this.filterTeamByOrg(v)) {
+          continue;
         }
-      })();
-
-      for (const t of this.teams) {
-        t.reset();
 
-        t.problemStatistics = this.contest.problems.map((p) => {
-          const ps = new TeamProblemStatistics();
-          ps.problem = p;
-          ps.contestPenalty = this.contest.penalty;
+        this.teams.push(v);
+      }
+    })();
 
-          return ps;
-        });
+    for (const t of this.teams) {
+      t.reset();
 
-        t.problemStatisticsMap = new Map(t.problemStatistics.map(ps => [ps.problem.id, ps]));
-      }
+      t.problemStatistics = this.contest.problems.map((p) => {
+        const ps = new TeamProblemStatistics();
+        ps.problem = p;
+        ps.contestPenalty = this.contest.penalty;
 
-      this.contest.problems.forEach((p) => {
-        p.statistics.reset();
+        return ps;
       });
 
+      t.problemStatisticsMap = new Map(t.problemStatistics.map(ps => [ps.problem.id, ps]));
+    }
+
+    this.contest.problems.forEach((p) => {
+      p.statistics.reset();
+    });
+  }
+
+  buildRank() {
+    (() => {
+      this.cleanRank();
+
       this.teams.forEach(t =>
         t.placeChartPoints.push({
           timePoint: 0,
@@ -187,6 +196,7 @@ export class Rank {
       (() => {
         this.rankStatistics.reset();
         this.rankStatistics.teamSolvedNum = Array(this.contest.problems.length + 1).fill(0);
+        this.rankStatistics.teamSolvedNumIndex = Array(this.contest.problems.length + 1).fill(0);
       })();
 
       let preSubmissionTimestampToMinute = 0;
@@ -295,6 +305,17 @@ export class Rank {
         this.rankStatistics.teamSolvedNum[t.solvedProblemNum]++;
       }
 
+      {
+        let current = 0;
+        const teamSolvedNum = this.rankStatistics.teamSolvedNum;
+        const teamSolvedNumIndex = this.rankStatistics.teamSolvedNumIndex;
+
+        for (let i = teamSolvedNumIndex.length - 1; i >= 0; i--) {
+          current += (teamSolvedNum[i] > 0 ? 1 : 0);
+          teamSolvedNumIndex[i] = current;
+        }
+      }
+
       if (this.teams.length > 0) {
         this.rankStatistics.maxSolvedProblems = this.teams[0].solvedProblemNum;
       }
@@ -394,4 +415,44 @@ export class Rank {
       s.timestamp <= this.options.timestamp,
     ).sort(Submission.compare);
   }
+
+  buildBalloons() {
+    this.balloons = [];
+    this.cleanRank();
+
+    const allSubmissions = this.getSubmissions();
+
+    for (let ix = 0; ix < allSubmissions.length; ix++) {
+      const s = allSubmissions[ix];
+
+      const teamId = s.teamId;
+      const problemId = s.problemId;
+      const team = this.teamsMap.get(teamId);
+      const problem = this.contest.problemsMap.get(problemId);
+
+      (() => {
+        if (team === undefined || problem === undefined) {
+          return;
+        }
+
+        const problemStatistics = team.problemStatisticsMap.get(problemId) as TeamProblemStatistics;
+
+        if (problemStatistics.isSolved) {
+          return;
+        }
+
+        if (s.isAccepted()) {
+          problemStatistics.isSolved = true;
+          problemStatistics.solvedTimestamp = s.timestamp;
+
+          const b = new Balloon();
+          b.team = team;
+          b.problem = problem;
+          b.submission = s;
+
+          this.balloons.push(b);
+        }
+      })();
+    }
+  }
 }