#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <unordered_map>
using namespace std;
#define VAR(n,v) typeof(v) n=(v)
#define REP(i,n) for(int i=1; i<=(n); ++i)
#define FOR(i,a,b) for(VAR(i,a); i!=(b); ++i)
#define FORE(it,t) FOR(it,t.begin(),t.end())
typedef vector<int> vi;
#define pb push_back
typedef unordered_map<int,int> hii;
typedef pair<int,int> pii;
#define mp make_pair
#define st first
#define nd second
typedef long long ll;
#define INF 1000000001
#define sz size()
#define ALL(t) (t).begin(),(t).end()
#define SC(a) scanf("%d", &a)
#define GET(a) int a; SC(a)
#define ISDEBUG 0
#define dprintf(...) if(ISDEBUG) \
{printf("\033[31m"); printf(__VA_ARGS__); printf("\033[0m");}
template <class It> void dptab(It b, It e, const char* f="%d ")
{if(ISDEBUG) {for(It it=b; it!=e; ++it) dprintf(f, *it); dprintf("\n"); }}
int n;
struct query {
int i,j,c;
bool operator<(const query& q) const {
return c < q.c;
}
};
vector<query> queries;
hii b2e, e2b;
void insert(int i, int j) {
b2e[i] = j;
e2b[j] = i;
}
void remove(int i, int j) {
b2e.erase(i);
e2b.erase(j);
}
bool add(int i, int j) {
if(i==j) return false;
if(b2e.count(i)) {
int mn = min(b2e[i], j);
int mx = max(b2e[i], j);
remove(i, b2e[i]);
insert(i, mn);
return add(mn, mx);
}
else if(e2b.count(j)) {
int mn = min(e2b[j], i);
int mx = max(e2b[j], i);
remove(e2b[j], j);
insert(mx, j);
return add(mn, mx);
}
else {
insert(i, j);
return true;
}
}
int main() {
SC(n);
REP(i, n) FOR(j,i,n+1) {
GET(c);
queries.pb({i,j+1,c});
}
sort(ALL(queries));
int cost = 0;
for(query& q: queries) {
if(!n) break;
if(add(q.i, q.j)) {
--n;
cost += q.c;
dprintf("used query\t%d-%d: %d\n", q.i, q.j-1, q.c);
}
else
dprintf("refused\t%d-%d: %d\n", q.i, q.j-1, q.c);
for(auto kv: b2e)
dprintf("[%d,%d) ", kv.st, kv.nd);
dprintf("\n");
}
printf("%d\n", cost);
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | #include <cstdio> #include <cstdlib> #include <cstring> #include <algorithm> #include <vector> #include <queue> #include <stack> #include <set> #include <unordered_map> using namespace std; #define VAR(n,v) typeof(v) n=(v) #define REP(i,n) for(int i=1; i<=(n); ++i) #define FOR(i,a,b) for(VAR(i,a); i!=(b); ++i) #define FORE(it,t) FOR(it,t.begin(),t.end()) typedef vector<int> vi; #define pb push_back typedef unordered_map<int,int> hii; typedef pair<int,int> pii; #define mp make_pair #define st first #define nd second typedef long long ll; #define INF 1000000001 #define sz size() #define ALL(t) (t).begin(),(t).end() #define SC(a) scanf("%d", &a) #define GET(a) int a; SC(a) #define ISDEBUG 0 #define dprintf(...) if(ISDEBUG) \ {printf("\033[31m"); printf(__VA_ARGS__); printf("\033[0m");} template <class It> void dptab(It b, It e, const char* f="%d ") {if(ISDEBUG) {for(It it=b; it!=e; ++it) dprintf(f, *it); dprintf("\n"); }} int n; struct query { int i,j,c; bool operator<(const query& q) const { return c < q.c; } }; vector<query> queries; hii b2e, e2b; void insert(int i, int j) { b2e[i] = j; e2b[j] = i; } void remove(int i, int j) { b2e.erase(i); e2b.erase(j); } bool add(int i, int j) { if(i==j) return false; if(b2e.count(i)) { int mn = min(b2e[i], j); int mx = max(b2e[i], j); remove(i, b2e[i]); insert(i, mn); return add(mn, mx); } else if(e2b.count(j)) { int mn = min(e2b[j], i); int mx = max(e2b[j], i); remove(e2b[j], j); insert(mx, j); return add(mn, mx); } else { insert(i, j); return true; } } int main() { SC(n); REP(i, n) FOR(j,i,n+1) { GET(c); queries.pb({i,j+1,c}); } sort(ALL(queries)); int cost = 0; for(query& q: queries) { if(!n) break; if(add(q.i, q.j)) { --n; cost += q.c; dprintf("used query\t%d-%d: %d\n", q.i, q.j-1, q.c); } else dprintf("refused\t%d-%d: %d\n", q.i, q.j-1, q.c); for(auto kv: b2e) dprintf("[%d,%d) ", kv.st, kv.nd); dprintf("\n"); } printf("%d\n", cost); } |
English