poj1812

C++ TLE,G++可以过。
Problem : 1812 ( Count the Tetris )     Judge Status : Accepted
RunId : 2899931    Language : G++    Author : huwenbiao
Code Render Status : Rendered By HDOJ G++ Code Render Version 0.01 Beta
/***************************************************************\
*Author:Hu Wenbiao
*Created Time: Tue 31 Aug 2010 04:03:31 PM CST
*File Name: main.cpp
*Description:polya原理。大数乘法直接循环乘的,不知道反复平方法
*会不会快些
\***************************************************************/

//*========================*Head File*========================*\\

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
/*----------------------*Global Variable*----------------------*/

int
A[
3000],B[3000];
int
N,C;

//*=======================*Main Program*=======================*//
using namespace std;

void
mul(int* A,int c,int m){
//A*=c^m
while(m--){
for
(int i=
1;i<=A[0];i++){
A[i]*=c;
}

for
(int i=
1;i<=A[0];i++){
if
(A[i]>
9999){
A[i+
1]+=A[i]/10000;
A[i]%=
10000;
}
}

if
(A[A[
0]+1])
A[
0]++;
}
}

void
add(int *B,int *A,int m){
//B+=A*m
B[0]=max(A[0],B[0]);
for
(int i=
1;i<=A[0];i++){
B[i]+=A[i]*m;
if
(B[i]>
9999){
B[i+
1]+=B[i]/10000;
B[i]%=
10000;
}
}

if
(B[B[
0]+1])
B[
0]++;
}

void
div(int*A,int*B){
//A=B/8
A[0]=B[0];
for
(int i=B[
0];i;i--){
A[i]=B[i]/
8;
B[i-
1]+=(B[i]%8)*10000;
}

if
(A[A[
0]]==0)
A[
0]--;
}

void
polya(int c,int n){
//polya主函数
memset(A,0,sizeof(A));
memset(B,
0,sizeof(B));
A[
0]=1;A[1]=1;
//旋转置换
if(n&1){//odd
mul(A,c,n*n/4+1);
add(B,A,
2);
mul(A,c,n*n/
4);
add(B,A,
1);
mul(A,c,n*n/
2);
add(B,A,
1);
}

else
{
//even
mul(A,c,n*n/4);
add(B,A,
2);
mul(A,c,n*n/
4);
add(B,A,
1);
mul(A,c,n*n/
2);
add(B,A,
1);
}

//翻转置换
memset(A,0,sizeof(A));
A[
0]=A[1]=1;
if
(n&
1){//odd
mul(A,c,(n*n-n)/2+n);
add(B,A,
4);
}

else
{
//even
mul(A,c,(n*n-n)/2+n);
add(B,A,
2);
memset(A,
0,sizeof(A));
A[
0]=A[1]=1;
mul(A,c,n*n/
2);
add(B,A,
2);
}

div(A,B);
}

int
main(){

//freopen("input","r",stdin);
while(scanf("%d%d",&N,&C)!=EOF){
if
(N==
1){
printf(
"%d\n",C);
continue
;
}

polya(C,N);
printf(
"%d",A[A[0]]);
A[
0]--;
while
(A[
0]){
printf(
"%.4d",A[A[0]]);
A[
0]--;
}

printf(
"\n");
}
}

你可能感兴趣的:(poj)