공부/SWEA

SWEA 8104. 조 만들기 (C++)

밤톨ㅇl 2024. 5. 11. 16:59
#include <stdio.h>
int N, K;
int group[21];
int main()
{
    int tc;
    scanf("%d", &tc);
    for (int t = 1; t <= tc; t++)
    {
        scanf("%d %d", &N, &K);

        for (int i = 1; i <= K; i++)
        {
            group[i] += i;
            for (int j = 2; j <= N; j += 2)
            {
                group[i] += (j * K) - i + 1;
            }
            for (int j = 3; j <= N; j += 2)
            {
                group[i] += ((j-1) * K) + i;
            }
        }
        printf("#%d ", t);

        for (int i = 1; i <= K; i++)
        {
            printf("%d ", group[i]);
        }
        printf("\n");
        for (int i = 1; i <= K; i++)
        {
            group[i] = 0;
        }
    }
    return 0;
}