Submission #1301023

#TimeUsernameProblemLanguageResultExecution timeMemory
1301023trandangquangReconstruction Project (JOI22_reconstruction)C++20
100 / 100
841 ms27656 KiB
#include<bits/stdc++.h> using namespace std; #define foru(i,a,b) for(int i=(a); i<=(b); ++i) #define ford(i,a,b) for(int i=(a); i>=(b); --i) #define rep(i,a) for(int i=0; i<(a); ++i) #define sz(a) (int)(a).size() #define all(a) (a).begin(),(a).end() #define bit(s,i) (((s)>>(i))&1) #define ii pair<int,int> #define vi vector<int> #define vii vector<ii> #define fi first #define se second #define ll long long #define int long long #define eb emplace_back #define pb push_back #define __builtin_popcount __builtin_popcountll #define _ << " " << template <class X, class Y> bool maxi(X &x, Y y){return x<y?x=y,true:false;} template <class X, class Y> bool mini(X &x, Y y){return x>y?x=y,true:false;} const int N=505; const int M=100001; int n,m,q; pair<ii,int> ed[M]; int par[N]; int find(int x){ return par[x]==0 ? x : par[x]=find(par[x]); } bool unite(int x, int y){ x=find(x);y=find(y); if(x!=y){ par[y]=x; return true; } return false; } int l[3*M],r[3*M]; pair<int,ii> cg[3*M]; void solve(){ cin>>n>>m; foru(i,1,m){ cin>>ed[i].fi.fi>>ed[i].fi.se>>ed[i].se; } sort(ed+1,ed+1+m,[](auto i, auto j){return i.se<j.se;}); memset(l,-0x3f,sizeof(l)); memset(r,0x3f,sizeof(r)); foru(i,1,m){ fill(par+1,par+1+n,0); ford(j,i-1,1){ unite(ed[j].fi.fi,ed[j].fi.se); if(find(ed[i].fi.fi)==find(ed[i].fi.se)){ l[i]=(ed[j].se+ed[i].se)/2+1; r[j]=(ed[j].se+ed[i].se)/2; break; } } } int sz=0; foru(i,1,m){ cg[++sz]={l[i],{-1,ed[i].se}}; cg[++sz]={ed[i].se,{2,-2*ed[i].se}}; cg[++sz]={r[i]+1,{-1,ed[i].se}}; } sort(cg+1,cg+1+sz); int j=1, s1=0, s2=0; cin>>q; foru(i,1,q){ int x; cin>>x; while(j<=sz && cg[j].fi<=x){ s1+=cg[j].se.fi; s2+=cg[j].se.se; ++j; } // cout<<s1 _ s2<<'\n'; cout<<1LL*s1*x+s2<<'\n'; } } int32_t main(){ #define task "test" if(fopen(task".inp", "r")){ freopen(task".inp", "r", stdin); freopen(task".out", "w", stdout); } cin.tie(0)->sync_with_stdio(0); int tc=1; //cin>>tc; foru(i,1,tc){ solve(); } }

Compilation message (stderr)

reconstruction.cpp: In function 'int32_t main()':
reconstruction.cpp:95:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   95 |         freopen(task".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
reconstruction.cpp:96:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   96 |         freopen(task".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...