Google面試題—有四個線程1、2、3、4。線程1的功能就是輸出1，線程2的功能就是輸出2，以此類推.........現在有四個文件ABCD

有四個線程1、2、3、4。線程1的功能就是輸出1，線程2的功能就是輸出2，以此類推.........現在有四個文件ABCD。初始都爲空。現要讓四個文件呈如下格式：A：1 2 3 4 1 2....B：2 3 4 1 2 3....C：3 4 1 2 3 4....D：4 1 2 3 4 1....請設計程序。

#include <stdio.h>   
#include <process.h>   
#include <windows.h> 
#include <fstream.h>
#include <string.h>


unsigned int __stdcall Fun(void *pPM);  

//線程個數
const int THREAD_NUM = 4;
const int FILE_NUM = 4;

ofstream ofile[FILE_NUM];

//某個線程的下一個要寫的文件
int FILE_THREAD[FILE_NUM]={0,1,2,3};
int NEXT_LOOP[FILE_NUM]={0,1,2,3,};

CRITICAL_SECTION g_csFile;

//循環次數
const int LOOP = 6;

//互斥事件
HANDLE g_hThreadEvent[THREAD_NUM];


int main()
{
	printf("\t有四個線程1、2、3、4。線程1的功能就是輸出1，線程2的功能就是輸出2\n");
	printf("以此類推.........現在有四個文件ABCD。初始都爲空。現要讓四個文件呈如下格式\n");

	int i = 0 , j;
	HANDLE hdl[THREAD_NUM];
	InitializeCriticalSection(&g_csFile);

	//事件開啓的時候是沒有觸發的
	for (i = 0; i < THREAD_NUM; i++) 
		g_hThreadEvent[i] = CreateEvent(NULL,FALSE,FALSE,NULL);

	//將線程編號作爲線程參數傳進去，先轉換爲指針，再轉換爲整形，
	
	char FILE_THREAD[] ="A.txt";

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

		FILE_THREAD[0] = i + 'A';
		ofile[i].open(FILE_THREAD,ios::trunc);

		if(ofile[i].fail())
		{
			printf("打開文件%s 失敗",FILE_THREAD);
			continue;
		}
	}

	for (i = 0; i < THREAD_NUM; i++) 
		hdl[i] = (HANDLE)_beginthreadex(NULL,0,Fun,(void*)i,0,NULL);


	//觸發第一個線程，線程函數會按順序自己觸發下一個要執行的線程
	SetEvent( g_hThreadEvent[0] );


	WaitForMultipleObjects(THREAD_NUM,hdl,TRUE,INFINITE);

	//清理
	for (i = 0; i < THREAD_NUM; i++)
	{
		CloseHandle(hdl[i]);
		CloseHandle(g_hThreadEvent[i]);
	}

	for( i  = 0 ; i< FILE_NUM;i++)
	{
		ofile[i].close();
	}
	DeleteCriticalSection(&g_csFile);

	return 0;
}

unsigned int __stdcall Fun(void *pPM)
{

	int num = (int)pPM ;
	int i = 0;
	for( i  = 0 ; i< LOOP;i++)
	{
		//等待順序觸發
		WaitForSingleObject(g_hThreadEvent[num],INFINITE);		
	
		EnterCriticalSection(&g_csFile);
	
		printf("  線程 %d 正在向%c文件寫入，下一次對文件%c操作的是線程%d\n",
			num + 1 , FILE_THREAD[num] + 'A',FILE_THREAD[num] + 'A',(num+1)%(THREAD_NUM) +1);

		ofile[FILE_THREAD[num]]<<num+1<<" ";

		Sleep(200);

		//記錄下一輪對應序號要操作的文件， (num+1)%(FILE_NUM) 事件對應的線程，操作文件FILE_THREAD[num]
		NEXT_LOOP[ (num+1)%(FILE_NUM) ] = FILE_THREAD[num] ;

		if( num + 1 == FILE_NUM)
		{
			printf("\n");
			//本輪寫入結束，將計算好的下一輪文件操作順序取過來
			memcpy(&FILE_THREAD,&NEXT_LOOP,FILE_NUM *sizeof(int));
		}
		
		LeaveCriticalSection(&g_csFile);	 
		
		//觸發下一個線程，1觸發2 ，2觸發3，3觸發4，4觸發1
		SetEvent(g_hThreadEvent[(num+1)%THREAD_NUM ]);
	}
	return 0;
}