51nod 1718 cos的多项式

题目链接

题目

小明对三角函数充满了兴趣，有一天他突然发现一个神奇的性质。
2cos(nx)似乎可以表示成2cos(x)的一个多项式。
但是小明并不能证明它的正确性。
现在给定n，问是否可以表示成这样的多项式，如果可以，只需输出各项系数的和。（Tip:如果这个和很大，那就高精度咯:)）
否则输出No

样例解释：2cos(3x)=(2cosx)^3-3*(2*cosx)，系数为1和-3，他们的和为-2。

题解

补充一下三角函数公式
cos(2a)=cos(a)*cos(a)-sin(a)*sin(a);
cos((k+1)*x)=cos(kx)*cos(x)-sin(kx)sin(x);
cos((k-1)*x)=cos(kx)*cos(-x)-sin(kx)*sin(-x)=cos(kx)cos(x)+sin(kx)sin(x);
2cos((k+1)x)+2cos((k+1)x)=2cos(kx)2cos(x)
f[k+1]+f[k-1]=f[k]*f[1];
f[1]=0;
f[k+1]=f[k]-f[k-1];
得出最后的递推方程为f[k+1]=f[k]-f[k-1];打表即可发现，本题具有循环节，由此即可解出题目。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=1e9+7;
const int N=1e5+10;
/*
    cos((k+1)*x)=cos(kx)*cos(x)-sin(kx)sin(x);
    cos((k-1)*x)=cos(kx)*cos(-x)-sin(kx)*sin(-x)=cos(kx)*cos(x)+sin(kx)sin(x);
    2*cos((k+1)*x)+2*cos((k+1)*x)=2*cos(kx)*2*cos(x)
    f[k+1]+f[k-1]=f[k]*f[1];
    f[1]=0;
    f[k+1]=f[k]-f[k-1];
*/
/*
    补充一下三角函数公式
    cos(2a)=cos(a)*cos(a)-sin(a)*sin(a);
*/
int main()
{
    ll n;
    scanf("%lld",&n);
    if(n%6==1 || n%6==5)
        printf("1\n");
    else if(n%6==2 || n%6==4)
        printf("-1\n");
    else if(n%6==3)
        printf("-2\n");
    else
        printf("2\n");

    return 0;
}