如何为VS Code配置类似LeetCode IDE的内存与时间限制？

VS Code本身没有内置的资源限制功能，但可以通过自定义脚本或系统命令模拟LeetCode、HackerRank这类OJ的时间/内存限制，以下是具体可行的方案：

一、Python代码的资源限制方案 #

编写一个包装脚本，用psutil库（跨平台）监控子进程的资源使用，替代默认的直接运行方式：

import os
import sys
import psutil
import subprocess
import time

def run_with_limits(script_path, time_limit=2, memory_limit=256):
    # 启动子进程执行目标代码
    proc = subprocess.Popen([sys.executable, script_path], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    start_time = time.time()
    max_memory = 0

    while proc.poll() is None:
        elapsed = time.time() - start_time
        # 检查超时
        if elapsed > time_limit:
            proc.kill()
            print(f"⚠️ 超时错误：超过{time_limit}秒限制")
            return
        # 检查内存占用（单位：MB）
        try:
            p = psutil.Process(proc.pid)
            current_mem = p.memory_info().rss / (1024 * 1024)
            max_memory = max(max_memory, current_mem)
            if max_memory > memory_limit:
                proc.kill()
                print(f"⚠️ 内存超限：超过{memory_limit}MB限制")
                return
        except psutil.NoSuchProcess:
            break
        time.sleep(0.1)

    # 输出程序结果
    stdout, stderr = proc.communicate()
    print("\n✅ 程序输出：")
    print(stdout.decode())
    if stderr:
        print("\n❌ 错误信息：")
        print(stderr.decode())
    print(f"\n📊 运行统计：最大内存 {max_memory:.2f}MB，耗时 {time.time() - start_time:.2f}秒")

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("用法：python limit_runner.py your_code.py")
        sys.exit(1)
    run_with_limits(sys.argv[1])

• 使用步骤：
  1. 安装psutil：pip install psutil
  2. 将上述代码保存为limit_runner.py放在项目根目录
  3. 运行目标代码时，执行python limit_runner.py your_code.py，可自行修改time_limit和memory_limit参数匹配OJ要求

二、编译型语言（C++/Java）的资源限制方案 #

Linux/macOS #

直接用系统内置的ulimit命令限制资源后运行程序：

• 限制CPU时间（比如2秒）：ulimit -t 2 && ./your_compiled_program
• 限制内存（比如256MB，单位KB）：ulimit -v 262144 && ./your_compiled_program

Windows #

建议通过WSL（Windows子系统Linux）使用上述ulimit命令，或者用PowerShell脚本监控进程资源：

$proc = Start-Process .\your_program.exe -NoNewWindow -PassThru
$startTime = Get-Date
$memoryLimitMB = 256
$timeLimitSec = 2

while (!$proc.HasExited) {
    $elapsed = (Get-Date) - $startTime
    if ($elapsed.TotalSeconds -gt $timeLimitSec) {
        $proc.Kill()
        Write-Host "⚠️ 超时错误：超过$timeLimitSec秒限制"
        exit
    }
    $memUsage = $proc.WorkingSet64 / (1024 * 1024)
    if ($memUsage -gt $memoryLimitMB) {
        $proc.Kill()
        Write-Host "⚠️ 内存超限：超过$memoryLimitMB MB限制"
        exit
    }
    Start-Sleep -Milliseconds 100
}

Write-Host "✅ 程序执行完成"

三、VS Code一键配置 #

在VS Code的.vscode/tasks.json中添加自定义任务，实现一键运行带资源限制：
以Python为例：

{
    "version": "2.0.0",
    "tasks": [
        {
            "label": "Run with OJ Limits",
            "type": "shell",
            "command": "python ${workspaceFolder}/limit_runner.py ${file}",
            "group": {
                "kind": "test",
                "isDefault": false
            },
            "problemMatcher": [],
            "presentation": {
                "reveal": "always",
                "panel": "new"
            }
        }
    ]
}

配置完成后，按Ctrl+Shift+B选择"Run with OJ Limits"，即可直接运行当前打开的代码并施加资源限制。

注意事项 #

• 模拟精度：脚本的监控是轮询式的，存在微小误差，但足够排查大部分超时/内存超限问题
• 平台差异：类Unix系统的资源限制工具更成熟，Windows建议优先用WSL
• OJ差异：不同OJ的资源计算逻辑可能略有不同（比如内存是否包含系统占用），模拟只能接近，无法完全一致

内容的提问来源于stack exchange，提问作者Raktim Majumdar