| # |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
| 934554 |
2024-02-27T14:57:49 Z |
8pete8 |
Robots (APIO13_robots) |
C++17 |
|
1 ms |
2396 KB |
#include<iostream>
#include<stack>
#include<map>
#include<vector>
#include<string>
#include<unordered_map>
#include <queue>
#include<cstring>
#include<cassert>
#include<limits.h>
#include<cmath>
#include<set>
#include<numeric> //gcd(a,b)
#include<algorithm>
#include<bitset>
#include<stack>
using namespace std;
#define ll long long
#define f first
#define endl "\n"
#define s second
#define pii pair<int,int>
#define pppiiii pair<pii,pii>
#define ppii pair<int,pii>
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define pb push_back
//#define mp make_pair
#define lb lower_bound
#define ub upper_bound
#define fastio ios::sync_with_stdio(false);cin.tie(NULL);
#pragma GCC optimize ("03,unroll-loops")
#define int long long
const int mod=1e9+7,mxn=505,lg=30,inf=1e18,minf=-1e18,Mxn=2e6,root=700;
void setIO(string name){
ios_base::sync_with_stdio(0); cin.tie(0);
freopen((name+".in").c_str(),"r",stdin);
freopen((name+".out").c_str(),"w",stdout);
}
pii pos[10];
char grid[mxn+10][mxn+10];
int n,h,w;
int dist[mxn+10][mxn+10][4];
vector<vector<int>>dp[10][10];
int realdist2[mxn+10][mxn+10];
int u[4]={1,0,-1,0},r[4]={0,-1,0,1};
bool vis[mxn+10][mxn+10];
bool valid(int x,int y){
if(x<=0||x>h||y<=0||y>w)return false;
if(grid[x][y]=='x')return false;
return true;
}
struct node{
int x,y,stp,di;
bool operator<(const node o)const{return stp>o.stp;}
};
vector<vector<int>> play(int ind){//get min dist
int x=pos[ind].f,y=pos[ind].s;
vector<vector<int>>realdist(h+1,vector<int>(w+1,inf));
for(int i=1;i<=h;i++)for(int j=1;j<=w;j++)for(int k=0;k<4;k++)dist[i][j][k]=inf;
priority_queue<node>q;
realdist[x][y]=0;
for(int i=0;i<4;i++)q.push({x,y,1,i});
while(!q.empty()){
auto [cx,cy,stp,di]=q.top();
q.pop();
if(dist[cx][cy][di]<=stp)continue;
dist[cx][cy][di]=stp;
if((grid[cx][cy]=='A'||grid[cx][cy]=='C')){
int add=1;
if(grid[cx][cy]=='A')add=-1;
di=(di+add+4)%4;
}
int nx=cx+u[di],ny=cy+r[di];
if(valid(nx,ny)){
if(dist[nx][ny][di]>stp)q.push({nx,ny,stp,di});
}
else{
realdist[cx][cy]=min(realdist[cx][cy],stp);
for(int k=0;k<4;k++)if(dist[cx][cy][k]>stp+1)q.push({cx,cy,stp+1,k});
}
}
return realdist;
}
void move(vector<vector<int>>&v){//move val
for(int i=1;i<=h;i++)for(int j=1;j<=w;j++)for(int k=0;k<4;k++)dist[i][j][k]=inf;
priority_queue<node>q;
for(int i=1;i<=h;i++)for(int j=1;j<=w;j++){
if(v[i][j]==inf)continue;
for(int k=0;k<4;k++)q.push({i,j,v[i][j]+1,k});
}
while(!q.empty()){
auto [cx,cy,stp,di]=q.top();
q.pop();
if(dist[cx][cy][di]<=stp)continue;
dist[cx][cy][di]=stp;
if((grid[cx][cy]=='A'||grid[cx][cy]=='C')){
int add=1;
if(grid[cx][cy]=='A')add=-1;
di=(di+add+4)%4;
}
int nx=cx+u[di],ny=cy+r[di];
if(valid(nx,ny)){
if(dist[nx][ny][di]>stp)q.push({nx,ny,stp,di});
}
else{
v[cx][cy]=min(v[cx][cy],stp);
for(int k=0;k<4;k++)if(dist[cx][cy][k]>stp+1)q.push({cx,cy,stp+1,k});
}
}
}
int32_t main(){
fastio
cin>>n>>w>>h;
int ans=inf;
for(int i=1;i<=9;i++)for(int j=1;j<=9;j++)dp[i][j].resize(h+1,vector<int>(w+1,inf));
for(int i=1;i<=h;i++)for(int j=1;j<=w;j++){
cin>>grid[i][j];
if(grid[i][j]>='1'&&grid[i][j]<='9')pos[grid[i][j]-'0']={i,j};
}
for(int i=1;i<=n;i++)dp[i][i]=play(i);
for(int i=1;i<=n;i++)for(int j=i+1;j<=n;j++){
for(int x=1;x<=h;x++)for(int y=1;y<=w;y++)dp[i][j][x][y]=dp[i][j-1][x][y]+dp[j][j][x][y];
move(dp[i][j]);
}
for(int x=1;x<=h;x++)for(int y=1;y<=w;y++){
for(int sz=1;sz<=n;sz++){
for(int l=1,r=sz;r<=n;l++,r++){
for(int k=l;k<r;k++)dp[l][r][x][y]=min(dp[l][r][x][y],dp[l][k][x][y]+dp[k+1][r][x][y]);
}
}
ans=min(ans,dp[1][n][x][y]);
}
cout<<ans;
/*
for(int i=1;i<=2;i++){
//break;
}*/
//for(int i=1;i<=h;i++)for(int j=1;j<=w;j++)ans=min(ans,dp[i][j][n]);
/*
1 2 3
*/
}
Compilation message
robots.cpp: In function 'void setIO(std::string)':
robots.cpp:37:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
37 | freopen((name+".in").c_str(),"r",stdin);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
robots.cpp:38:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
38 | freopen((name+".out").c_str(),"w",stdout);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
2396 KB |
Output is correct |
| 2 |
Incorrect |
1 ms |
2396 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
2396 KB |
Output is correct |
| 2 |
Incorrect |
1 ms |
2396 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
2396 KB |
Output is correct |
| 2 |
Incorrect |
1 ms |
2396 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
2396 KB |
Output is correct |
| 2 |
Incorrect |
1 ms |
2396 KB |
Output isn't correct |
| 3 |
Halted |
0 ms |
0 KB |
- |
