QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#312990#4423. AMPPZ in the times of disease_SheepsheepWA 9582ms6880kbC++141.9kb2024-01-24 15:18:552024-01-24 15:18:55

Judging History

你现在查看的是最新测评结果

  • [2024-01-24 15:18:55]
  • 评测
  • 测评结果:WA
  • 用时:9582ms
  • 内存:6880kb
  • [2024-01-24 15:18:55]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std ;
#define ll long long
const int inf = 1e8 ;
const int N = 3e6+9 ;
struct point
{
    int x , y , idx ;
};
point b[ 100 ] ;
int n , k , fa[ N ] , ans[N] ;
int Find( int x ){ return fa[x] == x ? x : fa[x] = Find( fa[x] ) ; }
long long dis( point a , point b )
{
    return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y) ;
}
void del()
{
    long long min_dis = (1ll<<63)-1 , idx_a = 0 , idx_b = 0 ;

    for( int i = 1 ; i <= k+1 ; i ++ )
    {
        long long md = (1ll<<63)-1 , idx = 0 ;
        for( int j = 1 ; j <= k+1 ; j ++ )
        {
            if( j == i ) continue ;
            if( dis(b[i],b[j]) < md )
            {
                md = dis(b[i],b[j]) ; idx = j ;
            }
        }
        if( md < min_dis )
        {
            min_dis = md ; idx_a = i ; idx_b = idx ;
        }
    }
    fa[Find(b[idx_a].idx)] = Find( b[idx_b].idx ) ;
    swap( b[idx_a] , b[k+1] ) ;

    //for( int i = 1 ; i <= n ; i ++ ) cout << Find(i) << " " ; cout << "\n" ;
}
void solve()
{
    cin >> n >> k ;
    queue<point>q ;
    for( int i = 1 ; i <= n ; i ++ )
    {
        point a ;
        cin >> a.x >> a.y ; a.idx = i ;
        if( i <= k+1 ) b[i] = a ;
        else q.push( a ) ;
        fa[i] = i ;
    }

    if( n == k )
    {
        for( int i = 1 ; i <= n ; i ++ ) cout << i << " " ; cout << "\n" ;
        return ;
    }

    del() ;
    while( !q.empty() )
    {
        b[k+1] = q.front() ; q.pop() ;
        del() ;
    }

    int base = 1 ;
    for( int i = 1 ; i <= n ; i ++ )
    {
        if( Find(i) == i ) ans[i] = base++ ;
    }

    for( int i = 1 ; i <= n ; i ++ ) cout << ans[ Find(i) ] << " " ;
    cout << "\n" ;
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int tt = 1 ; cin >> tt ;
    while( tt-- ) solve() ;
    return 0 ;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Wrong Answer
time: 9582ms
memory: 6880kb

input:

100
100000 20
270505 510725
245104 76414
131477 992924
781607 895592
562469 622976
564354 637966
980036 112090
522903 687218
113871 977855
6615 123673
13847 347618
657794 165707
420561 183995
11367 136391
507836 694877
985069 105115
774110 486921
14319 338715
774937 118145
981468 99089
803866 491315...

output:

1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

result:

wrong answer wrong solution (test case 1)