予鈴

アウトプットとメモ書きの中間みたいな記事がたくさん出ます。

AOJ Highway Express Bus

http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=0212

最短経路を求める問題ですが、道中にc枚割引券を使うことができます。
 d[vertex][cnt]として、拡張Dijkstraをやります。 cntは今持っている割引券の枚数です。
nowに対して、割引券を使う方と使わない方をqueにpushします。
なんだか拡張Dijkstraはプログラミングしてる感があるので好きです。

#include <vector>
#include <iostream>
#include <utility>
#include <algorithm>
#include <string>
#include <deque>
#include <queue>
#include <tuple>
#include <queue>
#include <functional>
#include <cmath>
#include <iomanip>
#include <map>
#include <set>
#include <numeric>
#include <unordered_map>
#include <unordered_set>
#include <complex>
#include <iterator>
#include <array>
#include <memory>
#include <stack>
#define vi vector<int>
#define vvi vector<vector<int> >
#define ll long long int
#define vl vector<ll>
#define vvl vector<vector<ll>>
#define vb vector<bool>
#define vc vector<char>
#define vs vector<string>
#define ld long double
#define INF 1e9
#define EPS 0.0000000001
#define rep(i,n) for(int i=0;i<n;i++)
#define loop(i,s,n) for(int i=s;i<n;i++)
#define all(in) in.begin(), in.end()
template<class T, class S> void cmin(T &a, const S &b) { if (a > b)a = b; }
template<class T, class S> void cmax(T &a, const S &b) { if (a < b)a = b; }
#define MAX 9999999
using namespace std;
typedef pair<int, int> pii;
typedef pair<double,double>pdd;
typedef pair<ll,ll>pll;
int main(){
    int c,n,m,s,e;
    while(cin>>c>>n>>m>>s>>e,c+n+m+e+s){
        --s,--e;
        vector<vector<pii>>edge(n,vector<pii>());
        rep(i,m){
            int a,b,cost; cin>>a>>b>>cost;
            --a,--b;
            edge[a].push_back(pii(b,cost));
            edge[b].push_back(pii(a,cost));
        }
        typedef  pair<pii,int>piii;
        priority_queue<piii,vector<piii>,greater<piii>>que;
        vector<vector<int>>d(n,vector<int>(11,INF));
        que.push(piii(pii(0,c),s));
        d[s][c]=0;
        while(!que.empty()){
            piii now=que.top(); que.pop();
            if(d[now.second][now.first.second]<now.first.first)continue;
            for(int i=0; i<edge[now.second].size();i++){
                pii next=edge[now.second][i];
                if(d[next.first][now.first.second]>d[now.second][now.first.second]+next.second){
                    d[next.first][now.first.second]=d[now.second][now.first.second]+next.second;
                    que.push(piii(pii(d[next.first][now.first.second],now.first.second),next.first));
                }
                if(now.first.second>0){
                    if(d[next.first][now.first.second-1]>d[now.second][now.first.second]+(next.second)/2){
                        d[next.first][now.first.second-1]=now.first.first+(next.second)/2;
                        que.push(piii(pii(d[next.first][now.first.second-1],now.first.second-1),next.first));
                    }
                }
            }
        }
        int ans=INF;
        for(int i=0; i<=c; i++)cmin(ans,d[e][i]);
        cout<<ans<<endl;
    }
}

ICPC参加記

大会前

いつのまにか自分のMacで参加することになってたので、環境整備係してました。

以前のチーム練習で、毎回include書くのはめんどくさいとわかったので、bitsを入れたりg++のバージョンを上げたりしてました。

チームメイトの素数うさぎ(@wk1080id)としゅもんくん(@shumon_84)はEmacsを使っていたので、とりあえずGNU Emacsを入れることに。

自分もEmacsになれるために、慣れないキーバインドを練習した結果、ctlキーの押し過ぎで小指を複雑骨折して2017年度のICPCは幕を閉じました。

来年は小指を丈夫にして挑みたいです。

 

 

 

 

 

 

 

 

 

 

 

 

※もちろんネタです。しかも自分はXcodeで殴りました。

 

個人的な準備としては、実装やるだけ問題をひたすら解いてました。(特にパズル系とシミュレーションやるだけ

 

 

 本番

(他の二人が詳しく書いてるの簡単に書きます。)

  • 始まってすぐに僕がA問題を解くことになってたので、とりあえずAC。模擬国内でWAったのでさっと通って安心しました。
  • この時の順位が34位だったらしくてちょっと嬉しかった。
  • 素数うさぎがBを実装しているときに、Cを読むと制約的に明らかにやるだけシミュレーションだったので、これバグらせたら本当にヤバイなと思いながら紙コーディング→サンプルが合わない。
  • 焦ったらだめだと言い聞かせながら、うさぎと実装を変わりつつデバッグしてなんとかAC。
  • Cが通って安心しました。
  • B問題は意味がわからないので、二人に任せっきりでした。
  • 他の問題はGがdfsかなぁって考えてたけど、いやまさかこんな後半に探索するだけなんて…って考えたりしましたが、他に解けそうな問題がなかったので実装するもTLEして終わりました。
  • 3完。99位。 ギリギリ2桁でした。

 

終わった後

ICPCが終わって肩の力が少し抜けました。

反省点・・・

  • 圧倒的なアルゴリズム力のNASA。勉強と精進が足りない。
  • 後輩に負けた。悲しみ(後輩は98位
  • 模擬国内の本番になって初めて提出方法を知って、チームに迷惑をかけたこと。
  • もう少し早くCを通したかった。

良かった点・・・

  • 実装力がほんの少しついたかなぁ。
  • 案外やるだけの問題が楽しいことに気づいたので、ICPC直前だけじゃなくて頑張りたい。

 

実は初めてのICPCでした。緊張感がすごい大会でした。チーム戦楽しい。

大会が終わった後は部室でお菓子パーティーやりました。

カタンを初めてやったのですが、後輩にボッコボコにやられたので次は勝ちたい。

 

ARC-B ツリーグラフ

B: ツリーグラフ - AtCoder Regular Contest 030 | AtCoder

n頂点からなるツリーグラフが与えられます。
いくつかの頂点にはお宝があり、そのお宝をすべて回収し、最初の頂点に戻ってくるまでの最小コストを求める問題です。

各頂点について考えます。
現在見ている頂点(vertex)を親に持つ頂点を通る必要がある→子孫のうち1つでも宝石を持っている。
という場合です。
宝石が見つかった場合は親にtrueを、そうでない場合はfalseを返すような再帰関数で解くことができます。

こういう問題が好きです()

#include <vector>
#include <iostream>
#include <utility>
#include <algorithm>
#include <string>
#include <deque>
#include <queue>
#include <tuple>
#include <queue>
#include <functional>
#include <cmath>
#include <iomanip>
#include <map>
#include <set>
#include <numeric>
#include <unordered_map>
#include <unordered_set>
#include <complex>
#include <iterator>
#include <array>
#include <memory>
#include <stack>
#define vi vector<int>
#define vvi vector<vector<int> >
#define ll long long int
#define vl vector<ll>
#define vvl vector<vector<ll>>
#define vb vector<bool>
#define vc vector<char>
#define vs vector<string>
#define ld long double
#define INF 1e9
#define EPS 0.0000000001
#define rep(i,n) for(int i=0;i<n;i++)
#define loop(i,s,n) for(int i=s;i<n;i++)
#define all(in) in.begin(), in.end()
template<class T, class S> void cmin(T &a, const S &b) { if (a > b)a = b; }
template<class T, class S> void cmax(T &a, const S &b) { if (a < b)a = b; }
#define MAX 9999999
using namespace std;
class Edge{
public:
    int to,from,cost;
};
vector<int>jem(101,0);
vector<vector<int>>edge(101);
vector<int>cost(101,0);
bool  dfs(int from,int vertex){
    bool var=false;
    if(edge[vertex].size()==1&&jem[vertex]==0)return false;
    rep(i,edge[vertex].size()){
        int next=edge[vertex][i];
        if(next==from)continue;
        if(dfs(vertex,next))var=true;
    }
    if(var||jem[vertex])cost[from]+=cost[vertex]+2;
    else return false;
    return true;
}
 
int main(){
    int  n,x; cin>>n>>x;
    rep(i,n){
        int foo; cin>>foo;
        if(foo)jem[i+1]=1;
    }
    rep(i,n-1){
        int to,from; cin>>to>>from;
        edge[to].push_back(from);
        edge[from].push_back(to);
    }
    rep(i,edge[x].size())dfs(x,edge[x][i]);
    cout<<cost[x]<<endl;
 
}

AOJ -0235 Sergent Rian

問題文:Sergeant Rian | Aizu Online Judge
すべての橋を爆破する最短コストは、最もコストが掛かる島(葉ではなく、1から到達するまでに最もコストがかかる島)を爆破する順番を最後にすれば良い。
サンプルを最短コストになるように鉛筆でなぞると、最もコストが掛かる島に辿り着くまでの橋以外は二度通ることになります。
したがって解法は

すべての橋を爆破するコスト*2-最長のコスト

となります。

#include <vector>
#include <iostream>
#include <utility>
#include <algorithm>
#include <string>
#include <deque>
#include <queue>
#include <tuple>
#include <queue>
#include <functional>
#include <cmath>
#include <iomanip>
#include <map>
#include <set>
#include <numeric>
#include <unordered_map>
#include <unordered_set>
#include <complex>
#include <iterator>
#include <array>
#include <memory>
#include <stack>
#define vi vector<int>
#define vvi vector<vector<int> >
#define ll long long int
#define vl vector<ll>
#define vvl vector<vector<ll>>
#define vb vector<bool>
#define vc vector<char>
#define vs vector<string>
#define ld long double
#define INF 1e9
#define EPS 0.0000000001
#define rep(i,n) for(int i=0;i<n;i++)
#define loop(i,s,n) for(int i=s;i<n;i++)
#define all(in) in.begin(), in.end()
template<class T, class S> void cmin(T &a, const S &b) { if (a > b)a = b; }
template<class T, class S> void cmax(T &a, const S &b) { if (a < b)a = b; }
#define MAX 9999999
using namespace std;
#define int ll
typedef pair<int, int> pii;
typedef pair<double,double>pdd;
class edge{
public:
    int to;
    int cost;
    void Init(){to=0; cost=0;}
    edge(int t, int c):to(t), cost(c){};
    
};
vi d(100,INF);
vector<edge>Edge[100];
void dijkstra(){
    d[1]=0;
    priority_queue<pii,vector<pii>,greater<pii> >que;
    que.push(pii(0,1));
    while(!que.empty()){
        pii now=que.top(); que.pop();
        int to_v=now.second;
        if(d[to_v]<now.first)continue;
        rep(i,Edge[to_v].size()){
            edge e=Edge[to_v][i];
            if(d[e.to]>d[to_v]+e.cost){
                d[e.to]=d[to_v]+e.cost;
                que.push(pii(d[e.to],e.to));
            }
        }
    }
}
signed main(){
    int n;
    while(cin>>n,n){
        vector<pair<pii,int>>data;
        rep(i,100)Edge[i].clear();
        rep(i,n-1){
            int from,to,cost;
            cin>>from>>to>>cost;
            Edge[from].push_back(edge(to,cost));
            Edge[to].push_back(edge(from,cost));
            data.push_back(make_pair(pii(from,to),cost));
        }
        int sum=0;
        rep(i,n+1)d[i]=INF;
        dijkstra();
        int maxi=0;
        for(int i=0; i<data.size();i++){
            if(data[i].first.first==1){
                if(Edge[data[i].first.second].size()==1)continue;
            }else if(data[i].first.second==1){
                if(Edge[data[i].first.first].size()==1)continue;
            }else if(Edge[data[i].first.first].size()==1 || Edge[data[i].first.second].size()==1)
                continue;
            sum+=data[i].second;
        }
        sum*=2;
        for(int i=1; i<=n;i++){
            if(i==1)continue;
            if(Edge[i].size()==1)continue;
            cmax(maxi, d[i]);
        }
        cout<<sum-maxi<<endl;
    }
}

D-Simple Knapsack

コンテストに参加しなかったのですが、想定解ではない方法で解いたので記事にします。

解法は,重さの制約が特殊なので最初の重さを引いて動的計画法をします。

dp[i][j][k] := j個目まで見た時の価値ik個選んだときの最大値
として、ナップサックのdpをやります。
多次元DPを初めてやったのですが、良い練習になったと思います。

#include <vector>
#include <iostream>
#include <utility>
#include <algorithm>
#include <string>
#include <deque>
#include <queue>
#include <tuple>
#include <queue>
#include <functional>
#include <cmath>
#include <iomanip>
#include <map>
#include <set>
#include <numeric>
#include <unordered_map>
#include <unordered_set>
#include <complex>
#include <iterator>
#include <array>
#include <memory>
#include <stack>
#define vi vector<int>
#define vvi vector<vector<int> >
#define ll long long int
#define vl vector<ll>
#define vvl vector<vector<ll>>
#define vb vector<bool>
#define vc vector<char>
#define vs vector<string>
#define ld long double
#define INF 1e9
#define EPS 0.0000000001
#define rep(i,n) for(int i=0;i<n;i++)
#define loop(i,s,n) for(int i=s;i<n;i++)
#define all(in) in.begin(), in.end()
template<class T, class S> void cmin(T &a, const S &b) { if (a > b)a = b; }
template<class T, class S> void cmax(T &a, const S &b) { if (a < b)a = b; }
#define MAX 9999999
using namespace std;
typedef pair<int, int> pii;
typedef pair<double,double>pdd;
typedef pair<ll,ll>pll;
#include<string.h>
ll dp[505][105][405]={0};
int main(){
    ll N,W; cin>>N>>W;
    vl v,w;
    ll lim=0;
    rep(i,N){
        ll value,wegiht;
        cin>>wegiht>>value;
        if(wegiht>W)continue;
        if(!i)lim=wegiht-1;
        dp[wegiht-lim][i][1]=value;
        v.push_back(value);
        w.push_back(wegiht-lim);
    }
    for(int j=0;j<N-1;j++){
        for(int i=0; i<401;i++){
            for(int k=0; k<=N;k++){
                if(dp[i][j][k]==0)continue;
                dp[i][j+1][k]=max(dp[i][j+1][k],dp[i][j][k]);
                if(i+lim*k+w[j+1]+lim>W)continue;
                cmax(dp[i+w[j+1]][j+1][k+1],dp[i][j][k]+v[j+1]);
            }
        }
    }
    ll ans=0;
    rep(i,401)rep(j,N+1)cmax(ans,dp[i][N-1][j]);
    cout<<ans<<endl;
}

Kansai Camp 参加記

3月26~3月29日にかけて、KMC(京都大学マイコンクラブ)とNAIST、RiPProで行った合同合宿の参加記です。

全部書いてると長くなるので、コンテストの内容とかはおいおい書くつもりです。

 

1日目

集合場所も知らず、準備も全くせずに11時頃に起きました。幸い15時*1集合だったので間に合いました…が、

 

 

集合場所で誰にも合流できずに15分ぐらい死んでました。

なんとかKMCの方と合流できたあと、民宿に移動 → 自己紹介 という流れに。

自己紹介で合宿参加者のAtcoder平均レートが1600とかいう異次元の数値*2であることが判明し、人権を喪失しました。

その流れでコドフォ(2時間ぐらい)→(めっちゃ豪華な)晩御飯→ABC

久々のコドフォだったので、(途中コンテスト以外の問題を解いてた) 色々忘れてました。

ちなみに、ABCは全完した人からお風呂に入れるみたいなノリがありました。*3

2日目 

朝風呂を無事ACして始まりました。 朝ごはんがめっちゃ美味しかったです。

その後は コドフォ→お昼→ARC →ARC →晩御飯→復習→コドフォ

という流れでした。

まぁコンテストは最下位かそれぐらいでした。悲しかった。

ただ、解いた問題で自分が知ってるアルゴリズムの問題はその日のうちに通せたのが唯一の進捗です。

休憩にcodefightsとかやってました。

最後のコドフォでは無限にWAを生産してました。

 

コドフォのサンプル弱すぎでは??

この日が一番コードを書いた気がします。

3日目

朝風呂ACしました。朝ごはんに2日連続で卵かけご飯が出たのが印象的でした。

この日は ARC(でチームを決める) → コドフォ(チーム戦)

みたいな流れです。

ARCの順位は…(察し)。典型DPを再帰で書いてTLEしたり、嘘解法を頑張って実装したりして悲しい事になりました。反省してます。

コドフォは4時間セットで、T.M先輩としょラーさん、KMCの方と4人でチームを組みました。

しょラーさんが解けそうな問題を僕に投げてくれて、2問ぐらい通しました。

ちなみにjudgeの時、WAしたら土下座しようと思ってました。

他の問題はさっぱりで、なんとか解こうとしたものの敗北。

4時間コンテストは流石に辛かったです。

晩御飯を食べた後はひたすら復習をしてました。(主にARC)

KMCの同年代のNさん*4

 

yebityon.hatenablog.com

 こういう解き方や、コンテストの別解だけではなく、セグメントツリーのライブラリや使い方など、本当に色々教えてもらいました。

とても勉強になりました、ありがとうございます。

余談ですが、けものフレンズの最終回を見ました。ハッピーエンドで終わった(?)ようで良かったです。

最終日

最終日は、競プロか周辺を散策するか迷って結局周辺を散歩しました。

近くの山を散歩したり、足湯に入って新歓の話やプログラミングの話をしてました。

足湯がめっちゃ気持ちよかったです。

 

まとめ

正直なところ、初日は合宿のレベルが高くて、自分は場違いでは?と思うこともありましたが、参加してよかったと思っています。

実装力もアルゴリズム力も無い僕ですが、周囲の方々のレベルが高い&競プロだけをする環境の中で、色々と感じることもあったし、何より競プロに真剣に取り組めたのではないかな、と思います。

合宿期間は健康とは決して言えない生活だったんですが、温泉パワーとご飯*5で乗り切れたのも、合宿ならではだと思いました。

合宿を企画してくださったKMCの皆さん、ありがとうございました。

 

 

 

*1:競プロerに優しい時間

*2:当社比

*3:もちろんできなかったのであとでasiさんに手伝ってもらってDを通しました。

*4:イニシャルです。

*5:三食ほんとに美味しかった

典型ナップサックを1次元配列で解く。

※一番PV数が多いので、ちょっと直しました(2020/5/24)

例題として、 B - 書き換え(Rewrite)を使います。
まずはこのコードを見てください。 

#include<bits/stdc++.h>
using namespace std;
int main(){
    int N,M;
    cin >> N >> M;
    vector<int>V(N),T(N);
    for(int i = 0; i < N; ++i){
        cin >> V[i] >> T[i];
    }
    vector<int>dp(M + 1,0);
    for(int i = 0; i < N; i++)
        for(int t = M; t >= 0;t--){
            if(t + T[i] <= M)
                dp[t + T[i]] = max(dp[t + T[i]], dp[t] + V[i]);
        }
    cout << *max_element(dp.begin(), dp.end()) << endl;
    
}

シンプルですね。時間計算量は O(NM)ですが、空間計算量は O(M)で少しお得です。

まずは2次元で考える。

二次元DPで考えると、
 dp[i][j] := i個目の書類を見ている時に,j時間使ったときの最大値
 dp[i + 1][j + T[i]]=max (dp[i + 1][j + T[i]],dp[i][j]+V[i]) という漸化式になります。


さて、この問題を1次元配列で解くことを考えます。
dp[i] :=時間iのときの最大の価値
つまり、どの書類を見ているか、という情報を捨てます。

以下が工夫するポイントです。

後ろ向きに更新する

時間 tを後ろからなめて、次のように更新します。

for(int t = M; t >= 0; --t)

 dp[t + T[i]] = max(dp[t + T[i]],dp[t] + v[i])
嬉しいポイントは、値の更新がすでに見た tに対して行われます。つまり、 i番目の書類を重複して更新されることがありません。

荷物を取らない推移ないけど?

必要無いからです。2次元DPの際は, i番目の書類まで見た結果を i + 1番目にも反映させる必要があるますが、この場合はありません。