Educational Codeforces Round 37 (Rated for Div. 2) D.Tank(思維題-dp/揹包+亂搞)

題目

你有n(2<=n<=5e3)個水池,初始第i個池子有ai(0<=ai<=1e5)的水且容積無限大。

一個容積爲k(1<=k<=5e3)的勺子,可以裝一個水池的min(k,a[i])的水放到其他水池。

勺子不能同時從兩個水池裏取水,從一個水池裏取了之後必須立刻倒掉,且只能倒在一個水池裏。

求能否配出V(0<=V<=1e9)容積的水,

要求輸出方案,第j行輸出三個數cnt x y,表示從x水池取水,倒向y水池,且取了cnt(1<=cnt<=1e9)勺

要求輸出的行數不超過n+5

思路來源

https://blog.csdn.net/afd5154/article/details/102254040

https://blog.csdn.net/tomjobs/article/details/104241859

題解

首先,如果可以不限次數的話,

一個水池用來存答案,剩下留一個水池存廢水,其他要麼倒向答案池要麼倒向廢水池

水的體積sum<v顯然不行,每個池子的餘量ai%k揹包湊不出v%k也顯然不行,

否則一定可以,先用若干個池子的餘量湊出v%k,這些池子中取一個當答案池,

由於sum>=v,整數v/k的部分用廢水池的水顯然能湊足,如果已經多了就反向倒出就行了

然後回來考慮這個n+5的次數,發現倒向答案池和廢水池最多n次,即完成揹包,

最後差k的整數倍,這個操作只用一次就可以解決了,所以n+5次數限制沒用

 

思路來源的兩個寫法,都判斷了哪些池是爲了湊齊揹包一定必需的

但我感覺沒必要判,搞出一種揹包方案就可以了,反正次數夠用

代碼

#include<bits/stdc++.h>
using namespace std;
const int N=5e3+10;
int n,k,v,a[N],sum;
bool dp[N][N];
vector<int>e[2];
bool ok(){
    if(sum<v)return 0;
    dp[0][0]=1;
    for(int i=1;i<=n;++i){
        for(int j=0;j<k;++j){
            dp[i][j]|=dp[i-1][((j-a[i])%k+k)%k];
            dp[i][j]|=dp[i-1][j];
        }
    }
    if(!dp[n][v%k])return 0;
    int now=v%k;
    //找到一組方案 使得e[0]內的數之和x%k 和v%k 同餘
    for(int i=n;i>=1;--i){
        if(dp[i-1][((now-a[i])%k+k)%k]){
            e[0].push_back(i);
            now=((now-a[i])%k+k)%k;
        }
        else{
            e[1].push_back(i);
        }
    }
    return 1;
}
int main(){
    memset(dp,0,sizeof dp);
    scanf("%d%d%d",&n,&k,&v);
    for(int i=1;i<=n;++i){
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    if(!ok())puts("NO");
    else{
        puts("YES");
        int now=0;
        for(int i=0;i<e[0].size();++i){
            now+=a[e[0][i]];
        }
        for(int i=1;i<e[0].size();++i){
            printf("%d %d %d\n",100000,e[0][i],e[0][0]);//e[0][i]->e[0][0]
        }
        for(int i=1;i<e[1].size();++i){
            printf("%d %d %d\n",100000,e[1][i],e[1][0]);//e[0][i]->e[0][0]
        }
        if(!e[0].size())e[0].push_back((e[1].back()));//除v%k==0情況外 不可能出現!e[0].size()的情形 但v%k==0時0<v 也需要湊齊體積爲v的水 找到一個空水箱存儲答案
        if(!e[1].size())e[1].push_back((e[0].back()));//所有水箱都用上了 那隨便找個放多餘水的空水箱
        if(now<v)printf("%d %d %d\n",(v-now)/k,e[1][0],e[0][0]);
        if(now>v)printf("%d %d %d\n",(now-v)/k,e[0][0],e[1][0]);
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章