NDOJ/judge/tasks/experiment.py

24 lines
687 B
Python
Raw Normal View History

2020-12-01 23:18:17 +00:00
from collections import defaultdict
from judge.models import SubmissionTestCase, Problem
def generate_report(problem):
testcases = SubmissionTestCase.objects.filter(submission__problem=problem).all()
score = defaultdict(int)
total = defaultdict(int)
rate = defaultdict(int)
for case in testcases.iterator():
score[case.case] += int(case.status == 'AC')
total[case.case] += 1
for i in score:
rate[i] = score[i] / total[i]
for i, _ in sorted(rate.items(), key=lambda x: x[1], reverse=True):
print(i, score[i], total[i], rate[i])
def test(code):
problem = Problem.objects.get(code=code)
generate_report(problem)