CodeForces505B【floyd】/C【記憶化搜索】

CodeForces505B

題意：

有M種不同顏色的線，給出這樣的圖，然後q個詢問，問兩點可以由多少種線連接。

思路：

其實這題很水，也是不必寫題解的。
但是呢！wa了兩遍woc…
寫這題主要也是看到了大哥曾經寫wa了。。然後他還沒補！（真是怠惰
然後我直接就是100次floyd，然後wa了！！！不講道理啊！這都wa！
然後實在想不出，也不想去打什麼BFS/DFS/並查集，然後就去題解看看唄，然後果然都是啥BFS/DFS/並查集這種做法，然後還是自己看看吧。。然後驚了！！拉了個floyd模板過了！
然後發現前面的floyd全部打錯了，哈哈哈哈。。。

代碼：

#include<stdio.h>
#include<string.h>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int Maxn = 1e2+10;

int edge[Maxn][Maxn][Maxn];
int n, m, q;

void floyd(int col){
    for(int k=1;k<=n;k++)
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    if((!edge[col][i][j]) && edge[col][i][k] && edge[col][k][j])
        edge[col][i][j] = edge[col][j][i] = 1;
}

int main(){
    int ans, u, v, c;
    memset(edge, 0, sizeof(edge));
    scanf("%d%d", &n, &m);
    for(int i=0;i<m;i++){
        scanf("%d%d%d", &u, &v, &c);
        edge[c][u][v] = edge[c][v][u] = 1;
    }
    for(int i=1;i<=m;i++) floyd(i);

    scanf("%d", &q);
    for(int i=0;i<q;i++){
        scanf("%d%d", &u, &v);
        ans = 0;
        for(int j=1;j<=m;j++){
            ans += edge[j][u][v];
        }
        printf("%d\n", ans);
    }
    return 0;
}

CodeForces505C

題意：

略

思路：

這個呀～
真是有點意思啊～
如果直接dp[cur_pos][pre_step],直接硬賽給pre_step，空間複雜度炸了。
因爲最差的d就是1，然後最多會跳幾次呢？(1+x)*x<=30000;
然後就很小呀～
那麼第二維就搞個相對值就好了，然後本地跑好像機子開了…三天沒關過機好慢…

代碼：

#include<stdio.h>
#include<string.h>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int INF = 0x3f3f3f3f;
const int Maxn = 3e4+10;

int p[Maxn];
int dp[Maxn][550];
int n, d;

int Mmax(int x,int y){
    return x > y ? x : y;
}
int change(int pos){
    return pos-d+250;
}

int DFS(int pos, int pre_step){
    int tmp;
    tmp = change(pre_step);
    if(dp[pos][tmp] != -1) return dp[pos][tmp];
    int new_step, temp = 0;

    new_step = pre_step;
    if(new_step+pos<=30000&&new_step>=1) temp = Mmax(DFS(pos+new_step, new_step), temp);
    new_step = pre_step-1;
    if(new_step+pos<=30000&&new_step>=1) temp = Mmax(DFS(pos+new_step, new_step), temp);
    new_step = pre_step+1;
    if(new_step+pos<=30000&&new_step>=1) temp = Mmax(DFS(pos+new_step, new_step), temp);

    dp[pos][tmp] = p[pos] + temp;
    return dp[pos][tmp];
}

int main(){
    int pos;
    scanf("%d%d", &n, &d);
    for(int i=0;i<n;i++){
        scanf("%d", &pos);
        p[pos]++;
    }

    memset(dp, -1, sizeof(dp));

    printf("%d", DFS(d, d));
    return 0;
}