Submission #255705

#	Time	Username	Problem	Language	Result	Execution time	Memory
255705		Namnamseo	Rope (JOI17_rope)	C++17	80 / 100	2572 ms	75384 KiB

rope

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pp;
void read(int& x){ scanf("%d",&x); }
template<typename T,typename... Args>
void read(T& a,Args&... b){ read(a); read(b...); }
#define all(x) (x).begin(),(x).end()
#define pb push_back

int n, m;
int col[1000010];

int ans[1000010];

vector<int> adj[1000010];

const int M=1048576;

void upd(int p, int v, int *t){
	p+=M; t[p]+=v; p>>=1;
	while(p) t[p]=max(t[p*2], t[p*2+1]), p>>=1;
}

int rM(int l, int r, int *t){
	int ret=0;
	l+=M; r+=M;
	while(l<=r){
		if(l&1) ret=max(ret, t[l++]);
		if((r&1)==0) ret=max(ret, t[r--]);
		l>>=1; r>>=1;
	}
	return ret;
}

void Do(){
	for(auto& v:adj) v.clear();
	int *self = new int[1000010]();
	int *t = new int[M<<1]();
    
    for(int i=1; i<=n; i+=2){
		if(i+1 <= n){
			int a=col[i], b=col[i+1];
			if(a == b){
				++self[a];
				upd(a, 2, t);
				continue;
			}
			adj[a].pb(b); adj[b].pb(a);
			upd(a, 1, t); upd(b, 1, t);
		} else upd(col[i], 1, t);
    }
    
    for(int i=1; i<=m; ++i){
		for(int x:adj[i]) upd(x, -1, t);
		int jib=rM(i, i, t); upd(i, -jib, t);
		ans[i]=min(ans[i],
			n-rM(1, m, t)-jib
		);
		for(int x:adj[i]) upd(x, 1, t);
		upd(i, jib, t);
    }

    delete[] self;
    delete[] t;
}

void Do2(){
	for(auto& v:adj) v.clear();
	int *self = new int[1000010]();
	int *t = new int[M<<1]();
    
    upd(col[1], 1, t);
    for(int i=2; i<=n; i+=2){
		if(i+1 <= n){
			int a=col[i], b=col[i+1];
			if(a == b){
				++self[a];
				upd(a, 2, t);
				continue;
			}
			adj[a].pb(b); adj[b].pb(a);
			upd(a, 1, t); upd(b, 1, t);
		} else upd(col[i], 1, t);
    }
    
    for(int i=1; i<=m; ++i){
		for(int x:adj[i]) upd(x, -1, t);
		int jib=rM(i, i, t); upd(i, -jib, t);
		ans[i]=min(ans[i],
			n-rM(1, m, t)-jib
		);
		for(int x:adj[i]) upd(x, 1, t);
		upd(i, jib, t);
    }

    delete[] self;
    delete[] t;
}

int main()
{
    read(n, m);
    for(int i=1; i<=n; ++i) read(col[i]);
    fill(ans+1, ans+m+1, n+1);
    Do(); Do2();
    reverse(col+1, col+n+1);
    Do(); Do2();
    for(int i=1; i<=m; ++i) printf("%d\n", ans[i]);
    return 0;
}

Compilation message (stderr)

rope.cpp: In function 'void read(int&)':
rope.cpp:5:25: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
 void read(int& x){ scanf("%d",&x); }
                    ~~~~~^~~~~~~~~

• Subtask 1: 15 / 15
• Subtask 2: 30 / 30
• Subtask 3: 10 / 10
• Subtask 4: 25 / 25
• Subtask 5: 0 / 20