MOve to Invert

[mailsatish]

!usr/bin/local/env python

import sys
j=0
d=int(raw_input())
for j in range(0,d):
n=int(raw_input())
if (n>2) :
sum=0
c=n-2
sum=(c*(c-1))/2
print sum+2
elif((n==1)or (n==2)):
print 1
elif(n==0):
print 0
j=j+1
sys.exit(0)

I am getting TLe..whats the prob with this code..

[nneonneo]

Again, there is a lot of input, and Python cannot process the input as fast as other languages.

One thing you can do is try
import psyco; psyco.full()
at the top of your code, and wrap the entire program in a main() function (make sure to call it at the end), like this:

import psyco; psyco.full()
def main():
    [program here]
main()

For this problem, it will not help, I think. There is just too much input.