八十七分求助。。。一个TLE
查看原帖
八十七分求助。。。一个TLE
689545
Simon_317楼主2022/7/25 15:27
#include<bits/stdc++.h>
using namespace std;
int flag[15],mp[15][15],n,ans;
bool judge(int x,int y){
	for(int i=1;i<x;i++){
		for(int j=1;j<=n;j++){
			if(mp[i][j]==1){
				if(abs(x-i)==abs(y-j)){
					return 0;
				}
			}
		}
	}
	return 1;
}
int prime(int a){
	int flag=1; 
	for(int i=2;i<a;i++){
		if(a%i==0) flag=0;
		
	}
	return flag;
}
void func(int h){
	if(h==n+1){
		ans++;
		if(ans<=3){
			for(int i=1;i<=n;i++){
				for(int j=1;j<=n;j++){
					if(mp[i][j]==1){
					cout<<j<<" ";
					}
				}
			}
			cout<<endl;
		}
		return;
	}
	
	
	for(int i=1;i<=n;i++){
		if(flag[i]==0&&judge(h,i)){
			flag[i]=1;
			mp[h][i]=1;
			func(h+1);
			flag[i]=0;
			mp[h][i]=0;
		}
	}
}
int main(){
	cin>>n;
	
	func(1);
	cout<<ans;
} 

测试样例全对,一个TLE

2022/7/25 15:27
加载中...