小米OJ-找多少个等差数列动态规划

Posted 2021-02-21 一路一歌

 

 

#include<stdio.h>
#include<string.h>
int arr[110];
int sum = 0;
int dp[201][402];
int num[201];
int main()
{
    int x;
    char ch;
    int len=0;
    while(~scanf("%d%c",&x,&ch)){
        num[len++] = x;
        if(ch!=\' \') break;
    };
    memset(dp,0,sizeof(dp));
    for(int k=-200;k<=200;k++)
        for(int i=0;i<len;i++)
            for(int j=i+1;j<len;j++)
                if(num[i]+k==num[j])
               dp[j][k+200]+=dp[i][k+200]+1;

    int ans=0;
    for(int i=0;i<len;i++)
        for(int k=0;k<=400;k++)
              ans+=dp[i][k];
    printf("%d\\n",ans);
    return 0;
}