QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#724900 | #9223. Data Determination | Okuchiri# | WA | 0ms | 3672kb | C++20 | 2.6kb | 2024-11-08 15:29:19 | 2024-11-08 15:29:21 |
Judging History
answer
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,k,x;
map<ll,ll>b;
ll a[200010];
map<ll,ll>q;
void solve()
{
cin>>n>>k>>x;
b.clear();
q.clear();
for(int i=1;i<=n;i++)
{
cin>>a[i],b[a[i]]=b[a[i]]+1;
}
sort(a+1,a+n+1);
for(int i=1;i<=n;i++)
{
q[a[i]]=q[a[i-1]]+1;
}
if(k%2==1)
{
if(b[x]==0)
{
cout<<"NIE\n";
return;
}
ll cnt1=0,cnt2=0;
for(int i=1;i<=n;i++)
{
if(a[i]<x)cnt1++;
if(a[i]>x)cnt2++;
}
ll tar=k/2,t=b[x]-1;
if(cnt1>cnt2)swap(cnt1,cnt2);
if(cnt1<tar&&cnt2<tar)
{
if(t+cnt1+cnt2>=2*tar)
{
cout<<"TAK\n";
return;
}
}
else if(cnt1<tar)
{
if(t+cnt1>=tar)
{
cout<<"TAK\n";
return;
}
}
else
{
cout<<"TAK\n";
return;
}
}
else
{
for(int i=1;i<=n;i++)
{
if(b[2*x-a[i]]==0)
{
b.erase(2*x-a[i]);
continue;
}
ll x1=a[i],y1=2*x-a[i];
if(x1!=y1)
{
ll cnt1=q[a[i]]-1;
ll cnt2=n-(q[y1]-b[y1])-1;
if(cnt1>=k/2-1&&cnt2>=k/2-1)
{
cout<<"TAK\n";
return;
}
}
else
{
ll cnt1=q[x]-b[x];
ll cnt2=n-q[x];
ll tar=k/2-1,t=b[x]-2;
if(b[x]<2)continue;
if(cnt1>cnt2)swap(cnt1,cnt2);
if(cnt1<tar&&cnt2<tar)
{
if(t+cnt1+cnt2>=2*tar)
{
cout<<"TAK\n";
return;
}
}
else if(cnt1<tar)
{
if(t+cnt1>=tar)
{
cout<<"TAK\n";
return;
}
}
else
{
cout<<"TAK\n";
return;
}
}
}
cout<<"NIE\n";
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
ll T = 1;
cin >> T;
while(T--)
{
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 0ms
memory: 3672kb
input:
3 6 4 42 41 43 41 57 41 42 4 2 4 1 2 5 8 7 5 57 101 2 42 5 57 7 13
output:
TAK NIE
result:
wrong answer 3rd lines differ - expected: 'NIE', found: ''