萌新刚学树状数组求条水题
查看原帖
萌新刚学树状数组求条水题
1330274
stdp楼主2025/1/15 11:12
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) x&(-(x))
using namespace std;
int t[250005],r[250005],n,ans1,ans2;
struct node{
	int num,val;
}a[250005];
bool cmp(node a,node b){
	if(a.val==b.val) return a.num<a.num;
	return a.val<b.val;
}
void add(int x,int k){
	for(int i=x;i<=n;i+=lowbit(i)){
		t[i]+=k;
	}
}
int sum(int x){
	int res=0;
	for(int i=x;i>=1;i-=lowbit(i)){
		res+=t[i];
	}
	return res;
}
signed main(){
	while(cin>>n){
		ans1=0,ans2=0;
		memset(t,0,sizeof t);
		memset(r,0,sizeof r);
		for(int i=1,j=1;i<=n*n;i++){
			cin>>a[j].val;
			a[j].num=j;
			if(a[j].val!=0) j++;
		}
		sort(a+1,a+n*n,cmp);
		for(int i=1;i<n*n;i++) r[a[i].num]=i;
		for(int i=n*n-1;i>=1;i--){
			add(r[i],1);
			ans1+=sum(r[i]-1);
		}
		memset(t,0,sizeof t);
		memset(r,0,sizeof r);
		for(int i=1,j=1;i<=n*n;i++){
			cin>>a[j].val;
			a[j].num=j;
			if(a[j].val!=0) j++;
		}
		sort(a+1,a+n*n,cmp);
		for(int i=1;i<n*n;i++) r[a[i].num]=i;
		for(int i=n*n-1;i>=1;i--){
			add(r[i],1);
			ans2+=sum(r[i]-1);
		} 
		if(ans1%2==ans2%2) cout<<"TAK\n";
		else cout<<"NIE\n";		
	}
	return 0;
}
2025/1/15 11:12
加载中...