全部超时了,请大佬们看一看,谢谢

P1464 Function

wkl20110514 @ 2022-12-21 08:48:55

全部超时了,请大佬们看一看,谢谢

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cmath>
#define LL long long
using namespace std;
LL a[3],b;
LL w(LL a,LL b,LL c){
    if(a<=0||b<=0||c<=0)return 1;
    else if(a>20||b>20||c>20)return w(20,20,20);
    else if(a<b&&b<c)return w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
    else return w(a-1,b,c)+w(a-1,b-1,c)+w(a-1,b,c-1)-w(a-1,b-1,c-1);
}
int main(){
    for(int i;i==i;){
        scanf("%lld%lld%lld",&a[0],&a[1],&a[2]);
        if(a[0]==-1&&a[1]==-1&&a[2]==-1)return 0;
        printf("w(%lld, %lld, %lld) = %lld\n",a[0],a[1],a[2],w(a[0],a[1],a[2]));
    }
}

好像输入50,50,50时时间就超了


by RP_INT_MAX @ 2022-12-21 09:04:54

@wkl20110514 不写记忆化的勇士


by Dr_Glitch @ 2022-12-21 09:08:14

你这个没有记忆化搜索啊,反复算肯定会超时,把w()改成这样就可以(记得定义f)

LL w(LL a,LL b,LL c){
    if(a<=0||b<=0||c<=0)return 1;
    if(a<=20&&b<=20&&c<=20&&f[a][b][c])return f[a][b][c];
    if(a>20||b>20||c>20)return w(20,20,20);
    if(a<b&&b<c)return f[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
    return f[a][b][c]=w(a-1,b,c)+w(a-1,b-1,c)+w(a-1,b,c-1)-w(a-1,b-1,c-1);
}

by wkl20110514 @ 2022-12-21 10:43:13

刚才学习了一番记忆化,懂了懂了,谢谢


by wkl20110514 @ 2022-12-21 11:52:17

@Dr_Glitch 请问这个三维数组应该开多大呢?ABC不会有负数么?怎么办呀?


by Dr_Glitch @ 2022-12-21 13:59:05

@wkl20110514 每一维开成20就可以了,前面的边界条件已经排除过负数的可能了呢(由于当数据大于20时会转化成小数据,所以需要记忆化的部分只有小于20的情况)


by wkl20110514 @ 2022-12-22 11:55:51

知道了,谢谢


|