#548975 (Java) No.880 Yet Another Segment Tree Problem

• 問題: No.880 Yet Another Segment Tree Problem
• ユーザー: uwi
• 提出日時: 2020-09-06 11:29:52
• 言語: Java (openjdk 23)
• 結果: AC
• 実行時間: 1,280 ms / 5,000 ms
• コード長: 7,866 bytes
• コンパイル時間: 3,486 ms
• コンパイル使用メモリ: 88,376 KB
• 実行使用メモリ: 65,516 KB
• 最終ジャッジ日時: 2025-06-20 10:35:29
• 合計ジャッジ時間: 26,711 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

ファイルパターン	結果
sample	AC * 1
other	AC * 38

ソースコード

package hackerearth.easy2009;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.InputMismatchException;

class D3 {
	InputStream is;
	PrintWriter out;
	String INPUT = "";
	
	void solve()
	{
		int n = ni(), m = ni();
		int[] a = na(n);
		SegmentTreeRGcdizeRUpdateRMaxQRSQ st = new SegmentTreeRGcdizeRUpdateRMaxQRSQ(a);
		for(int z = 0;z < m;z++){
			int t = ni();
			if(t == 1){
				int l = ni()-1;
				int r = ni()-1;
				int x = ni();
				st.update(l, r+1, x);
			}else if(t == 2){
				int l = ni()-1;
				int r = ni()-1;
				int x = ni();
				st.gcdize(l, r+1, x);
			}else if(t == 3){
				int l = ni()-1;
				int r = ni()-1;
				out.println(st.max(l, r+1));
			}else{
				int l = ni()-1;
				int r = ni()-1;
				out.println(st.sum(l, r+1));
			}
			/*
			tr(st.lcms);
			tr(st.covers);
			tr(st.sums);
			*/
		}
	}
	
	public static class SegmentTreeRGcdizeRUpdateRMaxQRSQ {
		public int M, H, N;
		public int[] maxs;
		public long[] sums;
		public long[] lcms;
		public int[] covers;
		private long I = Long.MAX_VALUE / 8;
		
		public SegmentTreeRGcdizeRUpdateRMaxQRSQ(int[] a)
		{
			N = a.length;
			M = Integer.highestOneBit(Math.max(N-1, 1))<<2;
			H = M>>>1;
			
			maxs = new int[M];
			sums = new long[M];
			lcms = new long[M];
			covers = new int[M];
			for(int i = 0;i < N;i++){
				sums[H+i] = lcms[H+i] = maxs[H+i] = covers[H+i] = a[i];
			}
			for(int i = H-1;i >= 1;i--)propagate(i);
		}
		
		private void propagate(int cur)
		{
			if(cur >= H){
				lcms[cur] = sums[cur] = maxs[cur] = covers[cur];
			}else{
				int L = 2*cur, R = 2*cur+1;
				if(covers[cur] == 0){
					lcms[cur] = lcm(lcms[L], lcms[R]);
					maxs[cur] = Math.max(maxs[L], maxs[R]);
					sums[cur] = sums[L] + sums[R];
				}else{
					int len = H/Integer.highestOneBit(cur);
					lcms[cur] = covers[cur];
					maxs[cur] = covers[cur];
					sums[cur] = (long)covers[cur] * len;
				}
			}
		}
		
		private long lcm(long a, long b)
		{
			if(a >= I || b >= I)return I;
			if(a == 0)return b;
			if(b == 0)return a;
			long ad = a / gcd(a, b);
			if((double)ad * b >= I)return I;
			return ad*b;
		}
		
		/*
		public static long gcd(long a, long b) {
			while (b > 0) {
				long c = a;
				a = b;
				b = c % b;
			}
			return a;
		}
		
		public static int gcd(int a, int b) {
			while (b > 0) {
				int c = a;
				a = b;
				b = c % b;
			}
			return a;
		}
		*/
		
		
		public static int gcd(int a, int b) {
			if(a == 0)return b;
			if(b == 0)return a;
			
			int ashift = Integer.numberOfTrailingZeros(a);
			int bshift = Integer.numberOfTrailingZeros(b);
			a >>>= ashift;
			b >>>= bshift;
			while(b != a){
				if(a > b){
					int t = a; a = b; b = t;
				}
				b -= a;
				b >>>= Integer.numberOfTrailingZeros(b);
			}
			
			return a<<Math.min(ashift, bshift);
		}
		
		public static long gcd(long a, long b) {
			if(a == 0)return b;
			if(b == 0)return a;
			
			int ashift = Long.numberOfTrailingZeros(a);
			int bshift = Long.numberOfTrailingZeros(b);
			a >>>= ashift;
			b >>>= bshift;
			while(b != a){
				if(a > b){
					long t = a; a = b; b = t;
				}
				b -= a;
				b >>>= Long.numberOfTrailingZeros(b);
			}
			
			return a<<Math.min(ashift, bshift);
		}
		
		public void gcdize(int l, int r, int v) {
			gcdize(l, r, v, 0, H, 1);
		}
		
		void push(int cur)
		{
			if(cur < H && covers[cur] != 0){
				// push
				for(int i = 2*cur;i <= 2*cur+1;i++){
					covers[i] = covers[cur];
					propagate(i);
				}
				covers[cur] = 0;
			}
		}
		
		boolean allsame(int cur)
		{
			return (long)maxs[cur] * (H/Integer.highestOneBit(cur)) == sums[cur];
		}
		
		protected void gcdize(int l, int r, int v, int cl, int cr, int cur)
		{
			if(v % lcms[cur] == 0){
				// not need to dig further
				return;
			}
			if(l <= cl && cr <= r){
				if(allsame(cur)){
					// all nodes have same value
					update(l, r, gcd(v, maxs[cur]), cl, cr, cur);
					return;
				}
			}
			push(cur);
			
			int mid = cl+cr>>>1;
			if(cl < r && l < mid)gcdize(l, r, v, cl, mid, 2*cur);
			if(mid < r && l < cr)gcdize(l, r, v, mid, cr, 2*cur+1);
			propagate(cur);
		}
		
		public void update(int l, int r, int v) {
			update(l, r, v, 0, H, 1);
		}
		
		protected void update(int l, int r, int v, int cl, int cr, int cur)
		{
			if(l <= cl && cr <= r){
				covers[cur] = v;
				propagate(cur);
				return;
			}
			
			push(cur);
			int mid = cl+cr>>>1;
			if(cl < r && l < mid)update(l, r, v, cl, mid, 2*cur);
			if(mid < r && l < cr)update(l, r, v, mid, cr, 2*cur+1);
			propagate(cur);
		}
		
		public long sum(int l, int r) {
			return sum(l, r, 0, H, 1);
		}
		
		protected long sum(int l, int r, int cl, int cr, int cur)
		{
			if(l <= cl && cr <= r){
				return sums[cur];
			}
			
			push(cur);
			
			long ret = 0;
			int mid = cl+cr>>>1;
			if(cl < r && l < mid)ret += sum(l, r, cl, mid, 2*cur);
			if(mid < r && l < cr)ret += sum(l, r, mid, cr, 2*cur+1);
			return ret;
		}
		
		public int max(int l, int r) {
			return max(l, r, 0, H, 1);
		}
		
		protected int max(int l, int r, int cl, int cr, int cur)
		{
			if(l <= cl && cr <= r){
				return maxs[cur];
			}
			
			push(cur);
			
			int ret = 0;
			int mid = cl+cr>>>1;
			if(cl < r && l < mid)ret = Math.max(ret, max(l, r, cl, mid, 2*cur));
			if(mid < r && l < cr)ret = Math.max(ret, max(l, r, mid, cr, 2*cur+1));
			return ret;
		}
	}
	
	void run() throws Exception
	{
		is = INPUT.isEmpty() ? System.in : new ByteArrayInputStream(INPUT.getBytes());
		out = new PrintWriter(System.out);
		
		long s = System.currentTimeMillis();
		solve();
		out.flush();
		if(!INPUT.isEmpty())tr(System.currentTimeMillis()-s+"ms");
	}
	
	public static void main(String[] args) throws Exception { new D3().run(); }
	
	private byte[] inbuf = new byte[1024];
	public int lenbuf = 0, ptrbuf = 0;
	
	private int readByte()
	{
		if(lenbuf == -1)throw new InputMismatchException();
		if(ptrbuf >= lenbuf){
			ptrbuf = 0;
			try { lenbuf = is.read(inbuf); } catch (IOException e) { throw new InputMismatchException(); }
			if(lenbuf <= 0)return -1;
		}
		return inbuf[ptrbuf++];
	}
	
	private boolean isSpaceChar(int c) { return !(c >= 33 && c <= 126); }
	private int skip() { int b; while((b = readByte()) != -1 && isSpaceChar(b)); return b; }
	
	private double nd() { return Double.parseDouble(ns()); }
	private char nc() { return (char)skip(); }
	
	private String ns()
	{
		int b = skip();
		StringBuilder sb = new StringBuilder();
		while(!(isSpaceChar(b))){ // when nextLine, (isSpaceChar(b) && b != ' ')
			sb.appendCodePoint(b);
			b = readByte();
		}
		return sb.toString();
	}
	
	private char[] ns(int n)
	{
		char[] buf = new char[n];
		int b = skip(), p = 0;
		while(p < n && !(isSpaceChar(b))){
			buf[p++] = (char)b;
			b = readByte();
		}
		return n == p ? buf : Arrays.copyOf(buf, p);
	}
	
	private char[][] nm(int n, int m)
	{
		char[][] map = new char[n][];
		for(int i = 0;i < n;i++)map[i] = ns(m);
		return map;
	}
	
	private int[] na(int n)
	{
		int[] a = new int[n];
		for(int i = 0;i < n;i++)a[i] = ni();
		return a;
	}
	
	private int ni()
	{
		int num = 0, b;
		boolean minus = false;
		while((b = readByte()) != -1 && !((b >= '0' && b <= '9') || b == '-'));
		if(b == '-'){
			minus = true;
			b = readByte();
		}
		
		while(true){
			if(b >= '0' && b <= '9'){
				num = num * 10 + (b - '0');
			}else{
				return minus ? -num : num;
			}
			b = readByte();
		}
	}
	
	private long nl()
	{
		long num = 0;
		int b;
		boolean minus = false;
		while((b = readByte()) != -1 && !((b >= '0' && b <= '9') || b == '-'));
		if(b == '-'){
			minus = true;
			b = readByte();
		}
		
		while(true){
			if(b >= '0' && b <= '9'){
				num = num * 10 + (b - '0');
			}else{
				return minus ? -num : num;
			}
			b = readByte();
		}
	}
	
	private static void tr(Object... o) { System.out.println(Arrays.deepToString(o)); }
}