Submission #959287

#	Time	Username	Problem	Language	Result	Execution time	Memory
959287		MarwenElarbi	Toll (BOI17_toll)	C++17	100 / 100	111 ms	86864 KiB

toll

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#define vi vector<int>
#define ve vector
#define ll long long
#define vl vector<ll>
#define vll vector<pair<ll,ll>>
#define onbit __builtin_popcount
#define ii pair<int,int>
#define vvi vector<vi>
#define vii vector<ii>
#define gii greater<ii>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define INF 1e18
#define eps 1e-7
#define eps1 1e-2
#define optimise ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
#define MAX_A 1e5+5
using namespace std;
using namespace __gnu_pbds;
template <class T>
using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const ll MOD = (1LL << 61) - 1;
const int nax = 1e5+5;
const int MAX_VAL = 1e6+1;
double PI=3.14159265359;
int arx[8]={1,0,0,-1,-1,-1, 1, 1};
int ary[8]={0,1,-1, 0, 1,-1,-1, 1};
void setIO(string s) {
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}
int k,n,m,o;
int dp[50005][17][5][5],ans[5][5],tmp[5][5];
void combine(int target[5][5], int a[5][5], int b[5][5]) {
    for (int x = 0; x < k; x++) {
        for (int y = 0; y < k; y++) {
            for (int z = 0; z < k; z++) {
                target[x][y] = min(target[x][y], a[x][z] + b[z][y]);
            }
        }
    }
}
int main(){
    optimise;
    //setIO("dec");
    cin>>k>>n>>m>>o;   
    memset(dp,0x3f,sizeof dp);
    for (int i = 0; i < m; ++i)
    {
        int a,b,t;
        cin>>a>>b>>t;
        dp[a/k][0][a%k][b%k] = t;
    }
    for (int j = 1; j < 17; ++j)
    {
        for (int i = 0; i + (1<<j) < (n+k-1)/k ; i++)
        {
            combine(dp[i][j],dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
        }
    }
    while(o--){
        int a,b;
        cin>>a>>b;
        memset(ans,0x3f,sizeof ans);
        for (int i = 0; i < 5; ++i) ans[i][i]=0;
        for (int cur = a/k, dis = b/k , i = 16 ; i >= 0; --i)
        {
            if(cur+(1<<i)<=dis){
                memset(tmp,0x3f,sizeof tmp);
                combine(tmp,ans,dp[cur][i]);
                memcpy(ans,tmp,sizeof ans);
                cur+=(1<<i);
            }
        }
        cout << (ans[a%k][b%k]==0x3f3f3f3f ? -1 : ans[a%k][b%k])<<'\n';
    }
    return 0;
}

Compilation message (stderr)

toll.cpp: In function 'void setIO(std::string)':
toll.cpp:36:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   36 |     freopen((s + ".in").c_str(), "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
toll.cpp:37:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   37 |     freopen((s + ".out").c_str(), "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 10 / 10
• Subtask 3: 8 / 8
• Subtask 4: 31 / 31
• Subtask 5: 44 / 44