#include <stdio.h>
#include <windows.h>
#include <conio.h>
#ifdef URLDownloadToFile
#undef URLDownloadToFile
#endif
typedef int(__stdcall *UDF)(LPVOID,LPCSTR,LPCSTR,DWORD,LPVOID);
UDF URLDownloadToFile = (UDF)GetProcAddress(LoadLibrary("urlmon.dll"),"URLDownloadToFileA");
char* strfind(char *text,char *temp)
{
int i = -1,j,l = strlen(temp);
while(text[++i])
{
for(j=0;j<l;j++)if(text[i+j] == 0 || text[i+j] != temp[j])break;
if(j == l)return text+i+l;
}
return 0;
}
void UTF8ToANSI(char *str)
{
int len = MultiByteToWideChar(CP_UTF8,0,str,-1,0,0);
WCHAR *wsz = new WCHAR[len+1];
len = MultiByteToWideChar(CP_UTF8,0,str,-1,wsz,len);
wsz[len] = 0;
len = WideCharToMultiByte(CP_ACP,0,wsz,-1,0,0,0,0);
len = WideCharToMultiByte(CP_ACP,0,wsz,-1,str,len,0,0);
str[len] = 0;
delete []wsz;
}
HANDLE hOutput;
char name[32];
int count[8];
void problem(char *str)
{
int i = 0,len;
DWORD unused;
char prob[32],url[128],*file,*ptr;
HANDLE hFile;
COORD pos = {0,2};
while(*str != '<')prob[i++] = *str++;
prob[i] = 0;
sprintf(url,"https://www.luogu.org/problemnew/show/%s",prob);
URLDownloadToFile(0,url,"download.tmp",0,0);
hFile = CreateFile("download.tmp",GENERIC_READ,FILE_SHARE_READ,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);
len = GetFileSize(hFile,0);
file = ptr = new char[len+3];
ReadFile(hFile,file,len,&unused,0);
file[len] = file[len+1] = 0;
CloseHandle(hFile);
file = strfind(file,"\xE9\x9A\xBE\xE5\xBA\xA6");
if(file == 0){delete []ptr;return;}
file = strfind(file,"lg-bg-");
if(file[0] == 'r')count[0]++;
else if(file[0] == 'o')count[1]++;
else if(file[0] == 'y')count[2]++;
else if(file[0] == 'g' && file[2] == 'e')count[3]++;
else if(file[0] == 'b' && file[4] == 'l')count[4]++;
else if(file[0] == 'p')count[5]++;
else if(file[0] == 'b' && file[4] == 'd')count[6]++;
else if(file[0] == 'g' && file[2] == 'a')count[7]++;
delete []ptr;
SetConsoleCursorPosition(hOutput,pos);
printf(
"%s 的统计: \n"
"入门难度: %d\n"
"普及-: %d\n"
"普及/提高-: %d\n"
"普及+/提高: %d\n"
"提高+/省选-: %d\n"
"省选/NOI-: %d\n"
"NOI/NOI+/CTSC: %d\n"
"尚无评定: %d\n"
"总数: %d\n\n\n\n\n",name,count[0],count[1],count[2],count[3],
count[4],count[5],count[6],count[7],count[0]+count[1]+count[2]+count[3]+count[4]+count[5]+count[6]+count[7]);
}
int main()
{
int uid,len,i = 0;
DWORD unused;
char url[128],user[16],*file,*ptr;
HANDLE hFile;
hOutput = GetStdHandle(STD_OUTPUT_HANDLE);
printf("请输入你的洛谷UID: ");
scanf("%d",&uid);
sprintf(url,"https://www.luogu.org/space/show?uid=%d",uid);
URLDownloadToFile(0,url,"download.tmp",0,0);
hFile = CreateFile("download.tmp",GENERIC_READ,FILE_SHARE_READ,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);
len = GetFileSize(hFile,0);
file = new char[len+3];
ReadFile(hFile,file,len,&unused,0);
file[len] = file[len+1] = 0;
CloseHandle(hFile);
UTF8ToANSI(file);
sprintf(user,"U%d ",uid);
ptr = strfind(file,user);
if(ptr)
{
while(ptr[0] != '<' || ptr[1] != '/' || ptr[2] != 'h')name[i++] = *ptr++;
printf("\n%s 的统计: ",name);
ptr = strfind(file,"通过题目</h2>\n[<");
if(ptr)while(*ptr != '<')
{
ptr = strfind(ptr,"\">");
problem(ptr);
ptr = strfind(ptr,"]\n");
}
else printf("还没有通过任何题目\n");
}
else printf("用户不存在\n");
DeleteFile("download.tmp");
delete []file;
printf("\nPress any key to exit. . .");
getchar();
return 0;
}