Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <bits/stdc++.h>
- using namespace std;
- int n, nr;
- unsigned long long p, a;
- int main()
- {
- ios_base::sync_with_stdio(0);
- cin.tie(0);
- cout.tie(0);
- cin>>n;
- for(int i=1; i<=n; ++i)
- {
- cin>>nr;
- if(nr%2)
- {
- a=(nr-1)/2;
- if(a%3==0)
- p=1ULL*2*(a/3)*(a+1)*(2*a+1) + 1ULL*nr*((nr+1)/2);
- else if((a+1)%3==0)
- p=1ULL*2*a*((a+1)/3)*(2*a+1) + 1ULL*nr*((nr+1)/2);
- else if((2*a+1)%3==0)
- p=1ULL*2*a*(a+1)*((2*a+1)/3) + 1ULL*nr*((nr+1)/2);
- else
- p=1ULL*2*a*(a+1)*(2*a+1)/3 + 1ULL*nr*((nr+1)/2);
- }
- else
- {
- a=nr/2;
- if(a%3==0)
- p=1ULL*2*(a/3)*(a+1)*(2*a+1);
- else if((a+1)%3==0)
- p=1ULL*2*a*((a+1)/3)*(2*a+1);
- else if((2*a+1)%3==0)
- p=1ULL*2*a*(a+1)*((2*a+1)/3);
- else
- p=1ULL*2*a*(a+1)*(2*a+1)/3;
- }
- cout<<p<<" ";
- }
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement