[2019CCPC哈爾濱] E Exchanging Gifts 拓撲排序

給出n1e6n\leq1e6個數字串,數字串的總和不會超過1e61e6,但是會有一個將兩個數字串連接的操作,總數字串的長度會比較大,最大會有1e181e18。然後求問你可以通過若干次交換,使得最多的數字自己的位置上和原來的數字不同,求這個最大值。
可以發現不同的數字不會超過1e61e6個,所以可以考慮基於值然後統計每個值的次數。對於每個被生成的串,由它向合成它的兩個串連一條有向邊,然後從最後一個串開始BFSBFS找到有效的那一部分子圖。然後對這張圖拓撲可以得到每個串被使用的次數,然後就容易統計每個數出現的次數了。
如果出現次數最多的那個數大於等於剩下的所有數的和,那麼答案就是剩下數的和的兩倍,否則都可以錯位。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=1e6+7;
inline char nc() {
	static char buf[1000000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
template <typename _Tp> inline void read(_Tp&sum) {
	char ch=nc();sum=0;
	while(!(ch>='0'&&ch<='9')) ch=nc();
	while(ch>='0'&&ch<='9') sum=(sum<<3)+(sum<<1)+(ch-48),ch=nc();
}
map<int,ll> mp; 
vector<int> a[N];
vector<int> go[N];
int in[N];
ll cnt[N]; 
bool vis[N];
void bfs(int s) {
	queue<int> q;
	q.push(s);
	vis[s]=1;
	while(!q.empty()) {
		int u=q.front();
		q.pop();
		for(auto &v:go[u]) {
			in[v]++;
			if(!vis[v]) {
				vis[v]=1;
				q.push(v);
			}
		}
	}
} 
void topo(int s) {
	queue<int> q;
	q.push(s);
	cnt[s]=1;
	while(!q.empty()) {
		int u=q.front();
		q.pop();
		if(go[u].size()==0) {
			for(auto &x:a[u]) { 
				mp[x]+=cnt[u];
			}
		}
		for(auto &v:go[u]) {
			cnt[v]+=cnt[u];
			if((--in[v])==0) q.push(v);
		}
	}
}
int main() {
	int T;
	read(T);
	while(T--) {
		int n;
		read(n);
		for(int i=1;i<=n;i++) {
			a[i].clear();
			go[i].clear();
			vis[i]=0; 
			in[i]=0;
			cnt[i]=0;
			mp.clear(); 
		}
		for(int i=1;i<=n;i++) {
			int opt,k,x,y;
			read(opt); 
			if(opt==1) {
				read(k);
				for(int j=1;j<=k;j++) {
					read(x); 
					a[i].emplace_back(x);
				}	
			}
			else {
				read(x);read(y); 
				go[i].emplace_back(x);
				go[i].emplace_back(y);
			}
		}
		bfs(n); 
		topo(n);
		ll sum=0,mx=0;
		for(auto &x:mp) {
			mx=max(mx,x.second);
			sum+=x.second;
		}
		if(mx>=sum-mx) printf("%lld\n",2ll*(sum-mx));
		else printf("%lld\n",sum); 
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章