UVa 12149 Feynman

Code：

#include<stdio.h>

#include<stdlib.h>

#include<iostream>

#include<algorithm>

#include<string.h>

using namespace std;

int arr[110],ans[110];

int fun(int n)

{

    if(n==1) return arr[1]=1;

    if(arr[n]!=0) return arr[n];

    return arr[n]=fun(n-1)+n*2-1;

}

int main()

{

    int n;

    memset(arr,0,sizeof(arr));

    memset(ans,0,sizeof(ans));

    fun(105);

    while(cin>>n && n!=0)

    {

        int sum=0;

        for(int i=1;i<=n;i++)

        {

            sum+=arr[i];

        }

        printf("%d\n",sum);

    }

    return 0;

}