#pragma GCC optimize ("O3,unroll-loops") #pragma GCC target ("avx,avx2,fma") #include #include #include #include //#define DEBUG // #define read(f,d,s) syscall(0,f,d,s) // #define write(f,d,s) syscall(1,f,d,s) #define gt(a,b) ((a)>(b)?(a):(b)) #define uintRd(v){ \ do{v=v*10+(*ptr++&0x0F);\ }while(*ptr&0x10);ptr++;\ } #define uintWr(src){ \ do *--ptr=(src%10)|'0'; \ while(src/=10); \ } #define bufSz (1<<14) uint8_t buf[bufSz], *ptr = buf, *end = buf+bufSz; uint32_t n,tmp,i,j,k; __m256i tmp256, mask256, max256; _Alignas(64) uint16_t max16[16]; _Alignas(64) __m256i indivAcc[64]; //_Alignas(64) uint16_t indivGrp[64][5][16]; // [batch cnt][grades][16 indivs] _Alignas(64) __m256i indivGrpX[64][5]; // [batch cnt][grades]([16 indivs]) _Alignas(64) int16_t groupCnt[5][9]; _Alignas(64) __m256i groupCntX[5][9]; _Alignas(64) __m256i groupNumX[9]; void print_m256i_16(__m256i *val){ uint16_t *v = (uint16_t*)val; for(int i=0;i<16;i++){ printf("%u ", v[i]); } putchar('\n'); } int main(){ //freopen("input.txt","r",stdin); for(i=0;i<9;i++) groupNumX[i] = _mm256_set1_epi16(i); read(0,buf,bufSz); uintRd(n); for(i=0;i>4][j]))[i&0xF] = tmp; //indivGrp[i>>4][j][i&0xF] = tmp; groupCnt[j][tmp]++; } } #ifdef DEBUG for(i=0;i<5;i++){ for(j=0;j<9;j++){ printf("%d\t",groupCnt[i][j]); } putchar(10); } #endif for(i=0;i<5;i++) for(j=0;j<9;j++) groupCntX[i][j] = _mm256_set1_epi16(groupCnt[i][j]); for(i=0;i<64;i++){ tmp256 = _mm256_setzero_si256(); for(j=0;j<5;j++){ for(k=0;k<9;k++){ tmp256= _mm256_add_epi16( _mm256_and_si256( _mm256_cmpeq_epi16( indivGrpX[i][j],groupNumX[k]), groupCntX[j][k]), tmp256); #ifdef DEBUG printf("i %d\tj %d\tk %d\n",i,j,k); print_m256i_16(&indivGrpX[i][j]); print_m256i_16(&groupNumX[k]); print_m256i_16(&groupCntX[j][k]); print_m256i_16(&tmp256); printf("\n"); #endif } } indivAcc[i] = tmp256; max256 = _mm256_max_epu16(max256, tmp256); break; } _mm256_store_si256((__m256i*)max16,max256); tmp = 0; for(i=0;i<16;i++) tmp = gt(tmp,max16[i]); #ifdef DEBUG printf("max: %d",tmp); printf("indivAcc\n"); for(i=0;i>4]))[i&0xF]); printf("\\indivAcc\n"); #endif for(i=0;i>4]))[i&0xF]==tmp)break; i++; ptr=end; uintWr(i); write(1,ptr,end-ptr); _Exit(0); }

제출 #143978

제출 정보

제출 결과

제출 코드

찾은 반례