Lähetit

input/code.cpp: In function 'void solve(std::vector<std::vector<bool> >, LL, LL)':
input/code.cpp:22:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(LL i = pos; i < positions.size(); i++){
                       ^

#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <algorithm>
#include <utility>

using namespace std;

typedef long long LL;

LL ans = 0;
LL n, k;
vector<pair<LL,LL> > positions;
    
void solve(vector<vector<bool> > available, LL pos, LL left){

    if(left == 0){
        ans++; return;
    }

    for(LL i = pos; i < positions.size(); i++){
        LL r = positions[i].first;
        LL c = positions[i].second;
        if(available[r][c]){
            vector<vector<bool> > available_new = available;
            LL r2 = r - c;
            for(LL c2 = 0; c2 < n; c2++, r2++){
                if(r2 >= 0 && r2 < n)
                    available_new[r2][c2] = false;
            }
            r2 = r + c;
            for(LL c2 = 0; c2 < n; c2++, r2--){
                if(r2 >= 0 && r2 < n)
                    available_new[r2][c2] = false;
            }
            
            solve(available_new, i + 1, left - 1);
        }
    }
}

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> k;    
    for(LL r = 0; r < n; r++){
        for(LL c = 0; c < n; c++){
            positions.push_back(make_pair(r,c));   
        }
    }
    vector<vector<bool> > available(n, vector<bool>(n,true));
    solve(available,0,k);
    
    cout << ans << endl;
    
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED