[C언어] 두 배열의 합집합, 교집합, 차집합 구하는 함수

#include <stdio.h>
#include <stdlib.h>
#define MAX_SET_SIZE 10
#define HAVE_ELEMENT 1
#define DO_NOT_HAVE_ELEMENT 0
int hasElement(int set[], int size, int element);
int setUnion(int set1[], int size1, int set2[], int size2, int setResult[]);
int setIntersecton(int set1[], int size1, int set2[], int size2, int setResult[]);
int setComplements(int set1[], int size1, int set2[], int size2, int setResult[]);
void printSet(int set[], int size);
int addOneElement(int set[], int size, int element);
int main(int argc, char *argv[])
{
     int i;
     int setA[MAX_SET_SIZE];
     int setB[MAX_SET_SIZE];
     int setC[MAX_SET_SIZE*2];
     int sizeA;
     int sizeB;
     int sizeC;
 
     printf("Enter the size of Set A:");
     scanf("%d",&sizeA);
 
     i = 0;
     while( i < sizeA )
     {
         printf("Enter the number for Set A (%d/%d):", i+1,sizeA );
         scanf( "%d", &setA[i] );
         i = addOneElement(setA, i, setA[i]);
     }
 
     printf("Enter the size of Set B:");
     scanf("%d",&sizeB);
 
     i=0;
     while( i < sizeB )
     {
         printf("Enter the number for Set B (%d/%d):", i+1, sizeB );
         scanf( "%d", &setB[i] );
         i = addOneElement(setB, i, setB[i]);
     }
 
     printf("Set A: ");
     printSet( setA, sizeA );
     printf("Set B: ");
     printSet( setB, sizeB );
     sizeC = setUnion( setA, sizeA, setB, sizeB, setC ); // Union, setC is the result set
     printf("Union of setA and setB: ");
     printSet( setC, sizeC );
     sizeC = setIntersecton( setA, sizeA, setB, sizeB, setC ); //Intersection, setC is the result set
     printf("Intersection of setA and setB: ");
     printSet( setC, sizeC );
     sizeC = setComplements( setA, sizeA, setB, sizeB, setC ); //Complements, setC is the result set
     printf("Set-theoretic difference of setA and setB (setA - setB): ");
     printSet( setC, sizeC );
 
     return 0;
}
 
int hasElement(int set[], int size, int element) //같은 원소가 존재하는지 확인
{
     int i = 0;
     for( i = 0; i < size; i++ )
     {
        if( set[i] == element )
            return HAVE_ELEMENT;
     }
     return DO_NOT_HAVE_ELEMENT;
}
// 원소를 집합에 추가. 이미 존재하면 추가하지 않고 redundant라고 출력한다. 현재의 집합 크기를 반환한다.
int addOneElement(int set[], int size, int element)
{
    if (hasElement(set, size, element) == HAVE_ELEMENT)
        printf("It is redundant. Please retry.\n");
    else
    {
        set[size] = element;
        size++;
    }
    return size;
        
}
void printSet(int set[], int size)
{
    int i;
    for( i = 0; i < size; i++ )
    {
        if (size == 1)
            printf("{ %d }\n", set[i]);
        else if (i == 0)
            printf("{ %d, ", set[i]);
        else if (i == size - 1)
            printf("%d }\n ", set[i]);
        else
            printf("%d, ", set[i]);
    }
}
int setUnion(int set1[], int size1, int set2[], int size2, int setResult[])//합
{
    int i, size3 = size1;
    for( i = 0; i < size1; i++ )
        setResult[i] = set1[i];
    for( i = 0; i < size2; i++ )
    {
        if (hasElement(setResult, size3, set2[i]) == DO_NOT_HAVE_ELEMENT)
        {
            setResult[size3] = set2[i];
            size3++;
        }
    }
    return size3;
}
int setIntersecton(int set1[], int size1, int set2[], int size2, int setResult[])//교
{
    int i, size3 = 0;
    if (size1 > size2)
    {
        for( i = 0; i < size2; i++ )
        {
            if(hasElement(set1, size1, set2[i]))
            {
                size3++;
                setResult[size3 - 1] = set2[i];
            }
        }
    }
    else
    {
        for( i = 0; i < size1; i++ )
        {
            if(hasElement(set2, size2, set1[i]))
            {
                size3++;
                setResult[size3 - 1] = set1[i];
            }
        }
    }
    return size3;
}
int setComplements(int set1[], int size1, int set2[], int size2, int setResult[])//차
{
    int i, size3 = 0;
    for( i = 0; i < size1; i++ )
    {
        if(hasElement(set2, size2, set1[i]) == DO_NOT_HAVE_ELEMENT)
        {
            size3++;
            setResult[size3 - 1] = set1[i];
        }
    }
 
    return size3;
} 

모든 함수는 새로운 배열 setResult의 size를 리턴한다.

 

처음엔 자꾸 쓰레기 숫자가 떠서 좀 짜증났는데 무튼 겨우 풀었다.

 

합,교,차 중 하나라도 할 수 있다면 나머지는 쉬울 것이다.