백준 3078번 문제 ( 좋은 친구 )

문제 출처 : https://www.acmicpc.net/problem/3078

3078번: 좋은 친구

이 문제는 큐와 배열을 선언하여서 해결한 문제이다.

전체 경우의 수를 돌아서 그 사이에 있는 글자 수들을 저장해서 해결해 나가는 문제이다.

아래는 해당 문제를 풀이한 소스이다.

#include<stdio.h>
#include<string.h>
#include<queue>
#include<iostream>
#pragma warning(disable:4996)
using namespace std;
int length[21];
queue <int> que;
long long int value = 0;
int cnt = 0;
struct A
{
    char arr[21];
    int size = 0;
}a;
int main()
{
    // freopen("test.inp", "r", stdin);
    // freopen("test.out", "w", stdout);
    int N;
    int K;
    scanf("%d %d\n", &N, &K);
    for (int i = 0; i < N; i++)
    {
        a.size = 0;
        scanf("%s", &a.arr);
        for (int j = 0;; j++)
        {
            if (a.arr[j] == '\0')
                break;
            else
                a.size++;
        }
        if (i >= K)
        {
            if (length[a.size] != 0)
            {
                cnt = length[a.size]; // 현재 str 
                value = value + cnt; // 값 더하고
                int minus = que.front(); // front의 글자수
                length[minus]--; // 
                que.pop();
                que.push(a.size);
                length[a.size]++;
            }
            else
            {
                int minus = que.front(); // front의 글자수
                length[minus]--; // 
                que.pop();
                que.push(a.size);
                length[a.size]++;
            }
        }
        else
        {
            if (length[a.size] != 0)
            {
                cnt = length[a.size];
                value = value + cnt;
            }
            que.push(a.size);
            length[a.size]++;
        }
    }
    printf("%lld", value);
}