#include <string>
#include <Windows.h>
#include <algorithm>

using namespace std;

int main()
{
WIN32_FIND_DATA find_data;
HANDLE hFind;
wstring extension[4] = {L"*.avi", L"*.mp4", L"*.mkv", L"*.wmv"};
wstring subex[3] = {L".smi", L".srt", L".ass"};
wstring files[256], subs[256];

int file_cnt = 0, sub_cnt = 0;

for(int i=0; i<4; ++i)
{
if((hFind = FindFirstFile(extension[i].c_str(), &find_data)) != INVALID_HANDLE_VALUE)
{
files[file_cnt++] = find_data.cFileName;
while(FindNextFile(hFind, &find_data))
files[file_cnt++] = find_data.cFileName;
break;
}
}
FindClose(hFind);
sort(&files[0], &files[file_cnt-1]);

int subtype = 0;
for(; subtype<3; ++subtype)
{
wstring temp = L"*" + subex[subtype];
if((hFind = FindFirstFile(temp.c_str(), &find_data)) != INVALID_HANDLE_VALUE)
{
subs[sub_cnt++] = find_data.cFileName;
while(FindNextFile(hFind, &find_data))
subs[sub_cnt++] = find_data.cFileName;
break;
}
}
FindClose(hFind);
sort(&subs[0], &subs[file_cnt-1]);

for(int i=0; i<file_cnt; ++i)
{
wstring temp = files[i].substr(0, files[i].rfind('.')) + subex[subtype];
MoveFile(subs[i].c_str(), temp.c_str());
}

return 0;
}