QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#106980#862. Social JusticeSortingWA 5ms3468kbC++201.9kb2023-05-20 00:47:222023-05-20 00:47:27

Judging History

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

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-05-20 00:47:27]
  • 评测
  • 测评结果:WA
  • 用时:5ms
  • 内存:3468kb
  • [2023-05-20 00:47:22]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std ;
typedef long long ll ;
typedef unsigned long long ull ;
typedef pair < int , int > pii ;
typedef vector < int > vi ;
#define fi first
#define se second
mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count());

const int MAXN = 2e5 + 7 ;

int n ;
pii a[ MAXN ] ;
ll pref[ MAXN ] ;
ll p , q ;
int add[ MAXN ] ;

bool f ( int len ) {
    for ( int i = 1 ; i + len - 1 <= n ; ++ i ) {
        ll sm = pref[ i + len - 1 ] - pref[ i - 1 ] ;
        ll aux1 = q * len * a[ i + len - 1 ].fi ;
        ll aux2 = p * sm ;
        if ( aux1 <= aux2 ) { return true ; }
    }
    return false ;
}

int bad[ MAXN ] ;

void solve ( ) {
    cin >> n ;
    for ( int i = 0 ; i <= n + 1 ; ++ i ) {
        add[ i ] = 0 ;
        bad[ i ] = 0 ;
    }
    for ( int i = 1 ; i <= n ; ++ i ) {
        cin >> a[ i ].fi ;
        a[ i ].se = i ;
    }
    cin >> p >> q ;
    sort ( a + 1 , a + n + 1 ) ;
    for ( int i = 1 ; i <= n ; ++ i ) {
        pref[ i ] = pref[ i - 1 ] + a[ i ].fi ;
    }
    int l , r , mid ;
    l = 1 , r = n ;
    while ( l < r ) {
        mid = ( l + r + 1 ) / 2 ;
        if ( f ( mid ) == true ) { l = mid ; }
        else { r = mid - 1 ; }
    }
    int len = l ;
    for ( int i = 1 ; i + len - 1 <= n ; ++ i ) {
        ll sm = pref[ i + len - 1 ] - pref[ i - 1 ] ;
        ll aux1 = q * len * a[ i + len - 1 ].fi ;
        ll aux2 = p * sm ;
        if ( aux1 <= aux2 ) {
            ++ add[ i ] , -- add[ i + len ] ;
        }
    }
    int sm = 0 , cnt = 0 ;
    for ( int i = 1 ; i <= n ; ++ i ) {
        sm = sm + add[ i ] ;
        if ( sm == 0 ) { bad[ a[ i ].se ] = 1 ; ++ cnt ; }
    }
    cout << cnt << "\n" ;
    for ( int i = 1 ; i <= n ; ++ i ) {
        if ( bad[ i ] == 1 ) { cout << i << " " ; }
    }
    cout << "\n" ;
}


int main ( ) {
    ios_base :: sync_with_stdio ( false ) ;
    cin.tie ( NULL ) ;
    int t = 1 ; cin >> t ;
    while ( t -- ) { solve ( ) ; }
    return 0 ;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3468kb

input:

3
4
1 2 3 4
3 2
5
1 15 2 5 1
2 1
5
1 2 3 1000 10000
4 3

output:

0

1
2 
2
4 5 

result:

ok 6 numbers

Test #2:

score: -100
Wrong Answer
time: 5ms
memory: 3392kb

input:

1000
1
10
3 2
2
10 100
3 2
3
10 10 100
3 2
4
1 2 3 4
3 2
5
1 15 2 5 1
2 1
5
1 2 3 1000 10000
4 3
6
1 2 3 4 1000 10000
4 3
5
50000 2 1 1 5000
2 1
10
1 15 2 5 1 10000 1 1 1 1
2 1
20
1 15 2 5 1 10000 1 1 1 1 1 15 2 5 1 10000 1 1 1 1
2 1
25
1 15 2 5 1 10000 1 1 1 1 1 15 2 5 1 10000 1 1 1 1 1 15 2 5 1
2 ...

output:

0

0

1
3 
0

1
2 
2
4 5 
3
1 5 6 
2
1 5 
3
2 4 6 
6
2 4 6 12 14 16 
8
2 4 6 12 14 16 22 24 
13
1 2 3 4 5 6 7 8 9 10 11 12 20 
15
1 2 3 4 5 6 7 8 9 10 11 12 18 19 20 
0

2
2 8 
2
9 12 
2
2 14 
10
1 4 5 6 7 13 14 15 18 20 
2
9 16 
2
16 18 
2
5 13 
10
4 5 6 9 11 15 16 18 19 20 
2
5 19 
2
5 13 
2
10 15...

result:

wrong answer 470th numbers differ - expected: '9', found: '10'