蒟蒻求调
查看原帖
蒟蒻求调
1008766
Sky_Aurora楼主2025/2/6 11:11

蒟蒻本来check用双重循环,87分TLE1个,之后就有了下面的代码:

#include<iostream>
using namespace std;

int a[15],n,cnt = 0;
bool vis[15];
int c[15],z[15],f[15];

bool check()
{
    for (int i = 1;i <= n;i++)
    {
        c[a[i]]++;
        z[i - a[i] + n]++;
        f[i + a[i]]++;
    }
    for (int i = 1;i <= 2 * n;i++)
        if (c[i] > 1 || z[i] > 1 || f[i] > 1) return false;
    return true;
}

void dfs(int step)
{
    if (step > n)
    {
        if (cnt >= 3)
        {
            cnt++;
            return ;
            
        }
        for (int i = 1;i <= n;i++) cout << a[i] << ' ';
        cout << endl;
        cnt++;
        return ;
    }
    for (int i = 1;i <= n;i++)
    {
        if (vis[i]) continue;
        a[step] = i;
        if (!check()) continue;
        vis[i] = true;
        dfs(step + 1);
        vis[i] = false;
        a[step] = 0;
    }
}

int main()
{
    cin >> n;
    dfs(1);
    cout << cnt << endl;
    return 0;
}

一重循环WA,悬关求调

2025/2/6 11:11
加载中...