C++--ACM之杭电OJ--2114Calculate S(n)

s(n)=1^3+2^3+3^3+………+n^3
s(n)的立方和公式为s(n)=(n*(n+1)/2)^2;
s(n)=1^2+2^2+3^3+……..+n^2
s(n)的平方和公式为s(n)=n*(n+1)*(2*n+1)/6;

#include<iostream>
using namespace std;
int main()
{
    long long n;         
    long long k;           //n与k类型要相同！！！！
    while(cin>>n)
    {
        k=((n*(n+1)/2)%10000)*((n*(n+1)/2)%10000);
        k=k%10000;
        if(k<10) cout<<"000"<<k<<endl;
        if(k>=10&&k<=99) cout<<"00"<<k<<endl;
        if(k>=100&&k<=999) cout<<"0"<<k<<endl;
        if(k>=1000) cout<<k<<endl;
    }
    return 0;
}