CF1512E-Permutation by Sum

题目链接

题目大意：给出一个数n，求它的一个排列，这个排列满足第l到r位的和的总和位s，求不出就输出-1

写得及其艰难的一道题。这是一道思维题，但想了很久没想到思维的方法，只能回溯暴搜，然后一直想剪枝的技巧，最后还是TLE了。

一个正确的思路是：先找出长度为l-r+1的最大子序列，然后依次将子序列中的最小数换成从1开始，更小的数，直到这个子序列的值等于s。这其中牵扯着一大堆判断，所以代码不短。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cctype>
using namespace std;
bool vis[505];
int main()
{
    int n, l, r, s, _, i, sum, fr;
    scanf("%d", &_);
    while (_--)
    {
        scanf("%d%d%d%d", &n, &l, &r, &s);
        sum = 0;
        for (i = 1; i <= n; i++)
            vis[i] = 0;
        stack<int> res;
        for (i = n; res.empty() || res.size() < r - l + 1; i--)
        {
            sum += i;
            res.push(i);
            vis[i] = 1;
        }
        fr = 1;
        queue<int> ans;
        while (sum > s)
        {
            if (res.empty())
                break;
            int t = res.top();
            res.pop();
            vis[t] = 0;
            if (t <= fr)
                break;
            if (sum - t + fr <= s)
            {
                ans.push(s - sum + t);
                vis[s - sum + t] = 1;
                sum = s;
                break;
            }
            sum = sum - t + fr;
            vis[fr] = 1;
            ans.push(fr++);
        }
        if (sum != s)
        {
            printf("-1\n");
            continue;
        }
        while (!res.empty())
        {
            ans.push(res.top());
            res.pop();
        }
        bool flag = 0;
        sum = 1;
        if (l > 1)
        {
            for (i = 1; i <= n; i++)
            {
                if (!vis[i])
                {
                    vis[i] = 1;
                    if (flag)
                        printf(" %d", i);
                    else
                    {
                        flag = 1;
                        printf("%d", i);
                    }
                    sum++;
                    if (sum == l)
                        break;
                }
            }
        }
        while (!ans.empty())
        {
            if (flag)
                printf(" %d", ans.front());
            else
            {
                flag = 1;
                printf("%d", ans.front());
            }
            ans.pop();
        }
        for (i = 1; i <= n; i++)
        {
            if (!vis[i])
            {
                if (flag)
                    printf(" %d", i);
                else
                {
                    flag = 1;
                    printf("%d", i);
                }
            }
        }
        printf("\n");
    }
    return 0;
}