회전 초밥(고등)

https://codeup.kr/problem.php?id=4752 

회전 초밥(고등)

1차원 배열에 현재 index의 초밥을 확인하며 테이블 배열에 더하거나 빼주면 된다.

 

#include <stdio.h>

int AN[3000001];

int TABLE[3001] = {
    0,
};

int IsOK(int type, int value)
{

    if (type)
    {

        // plus

        ++TABLE[value];

        if (TABLE[value] == 1)

            return 1;

        return 0;
    }

    else
    {

        // minus

        --TABLE[value];

        if (TABLE[value] == 0)

            return -1;

        return 0;
    }
}

int main()

{

    int N, d, k, c;

    int BN, bn = 0;

    int i;

    scanf("%d %d %d %d", &N, &d, &k, &c);

    // 0st input < bonus 'c' >

    bn += IsOK(1, c);

    for (i = 0; i < N; ++i)
    {

        scanf("%d", &AN[i]);

        // first input < 0 ~ k >

        if (i < k)

            bn += IsOK(1, AN[i]);
    }

    // initialize Biggest Number

    BN = bn;

    // second input < k ~ N+k >

    for (i = k; i < N + k; ++i)
    {

        if (i < N)
        {

            bn += IsOK(0, AN[i - k]);

            bn += IsOK(1, AN[i]);
        }

        else
        {

            bn += IsOK(0, AN[i - k]);

            bn += IsOK(1, AN[i - N]);
        }

        // compare Biggest Number

        if (BN < bn)

            BN = bn;
    }

    // answer

    printf("%d\n", BN);

    return 0;
}