PKU 3685. Matrix. [판정:WA]

#include <stdio.h> unsigned sigmaInc( unsigned ); unsigned sigmaDec( unsigned, unsigned ); unsigned sumOfGroup( unsigned, unsigned ); int rowIndex( unsigned, unsigned ); int columnIndex( unsigned, unsigned ); void main() { int trial; int i; unsigned dimension, searchOrder; scanf("%d", &trial); printf("\n"); /* skip a line */ for( i = 0; i < trial; i++ ) { unsigned j; unsigned beforeSum; double ithRow, jthCol; double wts; scanf("%u %u", &dimension, &searchOrder); for( j = 1; j < 2*dimension; j++ ) { beforeSum = sumOfGroup( j-1, dimension ); if( beforeSum<searchOrder && searchOrder<=sumOfGroup( j, dimension ) ) { /* searchOrder가 j번째 군에 속하는가를 판별 */ break; } } ithRow = rowIndex( j, dimension ); jthCol = columnIndex( j, dimension ); ithRow += ( searchOrder-( beforeSum+1 ) ); jthCol += ( searchOrder-( beforeSum+1 ) ); wts = ithRow*ithRow + 100000*ithRow + jthCol*jthCol - 100000*jthCol + ithRow*jthCol; printf("%.0f\n", wts); } } unsigned sigmaInc( unsigned n ) { /* 이 함수는 Σk (from 1 to n)을 구합니다 */ return ( n*( n+1 ) )/2; } unsigned sigmaDec( unsigned n, unsigned bound ) { /* 이 함수는 Σ(n+1-k) (from 1 to n)을 구합니다 */ return ( bound+1 )*n - ( n*( n+1 ) )/2; } unsigned sumOfGroup( unsigned k, unsigned dim ) { if( k <= dim ) { return sigmaInc(k); } else { return sigmaInc(dim) + sigmaDec( k-dim, dim-1 ); } } int rowIndex( unsigned n, unsigned dim ) { /* 이 함수는 j번째 군의 시작 원소의 행 번호를 구합니다 */ if( n <= dim ) { return 1; } else { return n-dim+1; } } int columnIndex( unsigned n, unsigned dim ) { /* 이 함수는 j번째 군의 시작 원소의 열 번호를 구합니다 */ if( n >= dim ) { return 1; } else { return dim-n+1; } }

젠장 -_-

한번에 AC받고 특집포스트 올리는게 목적이었는데

답만 나오고 판정은 WA

gg



아무도 솔루션을 안 올렸길래 내꺼 하나 던져봄 ㅇㅇ

다들 이런 알고리즘인거?