From b62fdbd18a8d007711802aca2826d5073ce03f19 Mon Sep 17 00:00:00 2001
From: klarkxy <278370456@qq.com>
Date: Fri, 1 Aug 2025 14:01:36 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E9=9A=BE=E5=BA=A6=E5=88=86?=
 =?UTF-8?q?=E7=B1=BB=E6=8C=89=E9=92=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 web/template/syzoj/problemset.php | 51 ++++++++++++++++++++++++++++++-
 1 file changed, 50 insertions(+), 1 deletion(-)

diff --git a/web/template/syzoj/problemset.php b/web/template/syzoj/problemset.php
index 7de31d9..130a0f4 100644
--- a/web/template/syzoj/problemset.php
+++ b/web/template/syzoj/problemset.php
@@ -15,6 +15,20 @@
             </div>
             <div class="results" style="width: 100%; "></div>
           </div>
+          
+          <!-- 添加难度按钮 -->
+          <div style="display: inline-block; margin-left: 10px; vertical-align: top;">
+            <?php for ($i = 1; $i <= 8; $i++): ?>
+              <button 
+                type="button" 
+                class="ui mini basic button difficulty-btn" 
+                data-difficulty="<?php echo $i; ?>"
+                style="margin-right: 5px; margin-bottom: 5px;"
+              >
+                难度<?php echo $i; ?>
+              </button>
+            <?php endfor; ?>
+          </div>
 
           <?php if (isset($_SESSION[$OJ_NAME . '_' . 'administrator'])): ?>
             <div class="ui search" style="width: 120px; height: 28px; margin-top: -5.3px; display: inline-block;">
@@ -72,6 +86,41 @@
     </div>
   </div>
 
+  <script>
+    // 添加难度按钮点击事件
+    document.querySelectorAll('.difficulty-btn').forEach(button => {
+      button.addEventListener('click', function() {
+        const difficulty = this.getAttribute('data-difficulty');
+        
+        // 创建隐藏表单提交
+        const form = document.createElement('form');
+        form.method = 'get';
+        form.action = '';
+        
+        // 添加难度参数
+        const input = document.createElement('input');
+        input.type = 'hidden';
+        input.name = 'difficulty';
+        input.value = difficulty;
+        form.appendChild(input);
+        
+        // 保留现有搜索参数
+        const existingSearch = document.querySelector('input[name="search"]');
+        if (existingSearch && existingSearch.value) {
+          const searchInput = document.createElement('input');
+          searchInput.type = 'hidden';
+          searchInput.name = 'search';
+          searchInput.value = existingSearch.value;
+          form.appendChild(searchInput);
+        }
+        
+        // 提交表单
+        document.body.appendChild(form);
+        form.submit();
+      });
+    });
+  </script>
+  
   <?php if (!isset($_GET['list'])) { ?>
 
     <div style="margin-bottom: 30px; ">
@@ -230,4 +279,4 @@
     <script type="text/javascript" src="include/jquery.tablesorter.js"></script>
 
   </div>
-  <?php include("template/$OJ_TEMPLATE/footer.php"); ?>
\ No newline at end of file
+  <?php include("template/$OJ_TEMPLATE/footer.php"); ?>